Added another test.

Added test checks if states will be loaded correctly.