Fixing error in work restarting code.