Fix the restart command