Fixed bug with screen states.