Fixed restart button not working