Fixing a minor issue with states.