Fixes states unit test