Fixed small bug in state data