diff --git a/src/Propellor/Property/Reboot.hs b/src/Propellor/Property/Reboot.hs index c262868..ac6f3a4 100644 --- a/src/Propellor/Property/Reboot.hs +++ b/src/Propellor/Property/Reboot.hs @@ -20,11 +20,8 @@ atEnd force resultok = property "scheduled reboot at end of propellor run" $ do return NoChange where atend r - | resultok r = liftIO $ - ifM (boolSystem "reboot" rebootparams) - ( return MadeChange - , return FailedChange - ) + | resultok r = liftIO $ toResult + <$> boolSystem "reboot" rebootparams | otherwise = do warningMessage "Not rebooting, due to status of propellor run." return FailedChange