diff --git a/mistral/tests/unit/engine/test_direct_workflow.py b/mistral/tests/unit/engine/test_direct_workflow.py
index d63ad8d5c..9d652f41d 100644
--- a/mistral/tests/unit/engine/test_direct_workflow.py
+++ b/mistral/tests/unit/engine/test_direct_workflow.py
@@ -1297,7 +1297,7 @@ class DirectWorkflowEngineTest(base.EngineTestCase):
             "join_task": 1
         }
 
-        self.assertDictContainsSubset(expected_context, output)
+        self._assert_dict_contains_subset(expected_context, output)
 
     def test_context_versioning(self):
         wf_text = """---