Fixed bug when game over