Fix states in tests.