Fixing state map to check incoming state.