Fix bug in states yaml