Attempt to fix stage change glitch