Fixed game over handling