Add restart to prevent restart service error