fix restart of game