Need to add test for states.Data

[jsgoecke]

Pull request came without a test, will need to add a test for it.