Fix small bug with the state propagation.