Restart button is now consistent