Fixed bug with restart command