Made ending a stage with the escape key working again.