Update test_state.cpp