diff --git a/substra/sdk/client.py b/substra/sdk/client.py
index 82bc273b..b8a8648c 100644
--- a/substra/sdk/client.py
+++ b/substra/sdk/client.py
@@ -1101,7 +1101,11 @@ def wait_function(
         asset_getter = self.get_function
         status_canceled = models.FunctionStatus.canceled.value
         status_failed = models.FunctionStatus.failed.value
-        statuses_stopped = (models.FunctionStatus.ready.value, models.FunctionStatus.canceled.value)
+        statuses_stopped = (
+            models.FunctionStatus.ready.value,
+            models.FunctionStatus.canceled.value,
+            models.FunctionStatus.failed.value,
+        )
         return self._wait(
             key=key,
             asset_getter=asset_getter,