Fix stage being stuck on playing