Added another test.

Added test checks if states will be loaded correctly.
Deleted obsolete comment.