Add test for state