Fixed restart  button implementation