diff --git a/package/tests/test_models/test_flowchart.py b/package/tests/test_models/test_flowchart.py
index eab528381..01238f286 100644
--- a/package/tests/test_models/test_flowchart.py
+++ b/package/tests/test_models/test_flowchart.py
@@ -1,7 +1,7 @@
 from functools import partial
 from pathlib import Path
 from textwrap import dedent
-from unittest.mock import Mock, call, patch
+from unittest.mock import call, patch
 
 import pytest
 from kedro.io import MemoryDataset
@@ -232,7 +232,7 @@ def test_create_single_parameter_with_complex_type(self):
         assert isinstance(parameters_node.parameter_value, str)
 
     def test_create_all_parameters_with_complex_type(self):
-        mock_object = Mock()
+        mock_object = object()
         parameters_dataset = MemoryDataset(
             data={
                 "test_split_ratio": 0.3,