1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp
index b07197797c..47af4a89b5 100644
--- a/src/armnn/test/CreateWorkload.hpp
+++ b/src/armnn/test/CreateWorkload.hpp
@@ -41,11 +41,13 @@ std::unique_ptr<Workload> MakeAndCheckWorkload(Layer& layer, Graph& graph, const
 }
 
 // Helper function to create tensor handlers for workloads, assuming they all use the same factory.
-void CreateTensorHandles(armnn::Graph& graph, armnn::IWorkloadFactory& factory)
+void CreateTensorHandles(armnn::Graph& graph,
+                         armnn::IWorkloadFactory& factory)
 {
+    TensorHandleFactoryRegistry tmpRegistry;
     for (auto&& layer : graph.TopologicalSort())
     {
-        layer->CreateTensorHandles(graph, factory);
+        layer->CreateTensorHandles(tmpRegistry, factory);
     }
 }