Fixed restart game bug