Fix assert failure in await expression with non-null assertion

Fix assert failure in await expression with non-null assertion #152