Update tests for state methods