Fix play state bug