changed next_state to account for state flow
updated failing test to account for change