diff --git a/tests/laborious/workflows/subworkflows/test_format_and_export_prediction.py b/tests/laborious/workflows/subworkflows/test_format_and_export_prediction.py
index 59e39b3..ecd3bc2 100644
--- a/tests/laborious/workflows/subworkflows/test_format_and_export_prediction.py
+++ b/tests/laborious/workflows/subworkflows/test_format_and_export_prediction.py
@@ -93,12 +93,12 @@ async def test_run_none_path_flag(workflow_mock, format_and_export_prediction):
                 {
                     'schema': input_data['schema'],
                     'table_name': input_data['table_name'],
-                    'data': workflow_mock.execute_local_activity_method.return_value,
-                    **metadata,
+                    'data': prediction_data,
                     'timestamp_conversion': {
                         'column': 'timestamp',
                         'format': DATETIME_FORMAT_WITH_TZ,
                     },
+                    **metadata,
                 },
                 retry_policy=ANY,
                 start_to_close_timeout=ANY,