fix restart game function