Fixed game over logic