Fixed state transition
Merge pull request #30 from google/states-fix

Fixed state transition