fixed issue with restartButton staying on screen