Fixed error in state propagation