It would be nice if the game kept the movement configurations throughout the stages

Agree, it is planned.