fixed restart game bug