fixed the restart button