fix restart and gameover