Fixing restart task