diff --git a/tests/ut/datavisual/processors/test_graph_processor.py b/tests/ut/datavisual/processors/test_graph_processor.py
index 1a263571..3c95989d 100644
--- a/tests/ut/datavisual/processors/test_graph_processor.py
+++ b/tests/ut/datavisual/processors/test_graph_processor.py
@@ -70,7 +70,7 @@ class TestGraphProcessor:
         self._train_id = log_dir.replace(summary_base_dir, ".")
 
         log_operation = LogOperations()
-        self._temp_path, self._graph_dict = log_operation.generate_log(PluginNameEnum.GRAPH.value, log_dir)
+        self._temp_path, self._graph_dict, _ = log_operation.generate_log(PluginNameEnum.GRAPH.value, log_dir)
         self._generated_path.append(summary_base_dir)
 
         self._mock_data_manager = data_manager.DataManager([DataLoaderGenerator(summary_base_dir)])
diff --git a/tests/utils/log_operations.py b/tests/utils/log_operations.py
index bc4e9e38..cd0d45a4 100644
--- a/tests/utils/log_operations.py
+++ b/tests/utils/log_operations.py
@@ -157,7 +157,7 @@ class LogOperations:
                 graph_dict = json.load(load_f)
 
             graph_dict = log_generator.generate_log(temp_path, graph_dict)
-            return temp_path, graph_dict
+            return temp_path, graph_dict, None
 
         steps_list = log_settings.get('steps', [1])
         tag_name = log_settings.get('tag', 'default_tag')