Fix states keyword

Fix error introduced on 70d4e3e9a178
review5371002