Add some testing for states.