Fix RestartCommand.cs