Fix restart button.