fixed problem with game restart