Update state_manager.go