Merge pull request #124 from innaDa/fixesStatesUnitTest

Fixes states unit test