diff --git a/mistral/tests/resources/data_flow/two_subsequent_tasks.yaml b/mistral/tests/resources/data_flow/two_subsequent_tasks.yaml
index 4db9e0138..d7cf9a39c 100644
--- a/mistral/tests/resources/data_flow/two_subsequent_tasks.yaml
+++ b/mistral/tests/resources/data_flow/two_subsequent_tasks.yaml
@@ -21,7 +21,8 @@ Namespaces:
         parameters:
           - full_name
         output:
-          greeting: $
+          greeting:
+            greet_message: $
 
 
 Workflow:
diff --git a/mistral/tests/unit/engine/test_data_flow.py b/mistral/tests/unit/engine/test_data_flow.py
index c6ecc62f7..fc9b254ae 100644
--- a/mistral/tests/unit/engine/test_data_flow.py
+++ b/mistral/tests/unit/engine/test_data_flow.py
@@ -304,10 +304,14 @@ class DataFlowTest(base.EngineTestCase):
                              build_greeting_task['parameters'])
         self.assertDictEqual(
             {
-                'greet_msg': 'Hello, John Doe!',
+                'greet_msg': {
+                    'greet_message': 'Hello, John Doe!'
+                },
                 'task': {
                     'build_greeting': {
-                        'greeting': 'Hello, John Doe!',
+                        'greeting': {
+                            'greet_message': 'Hello, John Doe!'
+                        },
                     }
                 }
             },