Fix bug state doesn't save for Pet