Trying to fix Restart