diff --git a/libs/sm/cleanup.py b/libs/sm/cleanup.py
index e75f0999f..963e435e0 100644
--- a/libs/sm/cleanup.py
+++ b/libs/sm/cleanup.py
@@ -2598,7 +2598,7 @@ def _checkSlaves(self, vdi):
                 raise AbortException("Aborting due to signal")
             try:
                 self._checkSlave(hostRef, vdi)
-            except util.CommandException:
+            except XenAPI.Failure:
                 if hostRef in onlineHosts:
                     raise