Fixed restart button ghost bug