Fixed game over bug