Fix state generator to evaluate the expression statement expression before resolving the current state.  This is important for example in await expressions where otherwise it would add the statement to the state before the await is complete.