diff --git a/src/backends/OutputHandler.cpp b/src/backends/OutputHandler.cpp
index c1be5b7..4dfa1a6 100644
--- a/src/backends/OutputHandler.cpp
+++ b/src/backends/OutputHandler.cpp
@@ -25,6 +25,11 @@
     m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo);
 }
 
+void OutputHandler::CreateTensorHandles(const IWorkloadFactory& factory, DataLayout dataLayout)
+{
+    m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, dataLayout);
+}
+
 void OutputHandler::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const
 {
     dataCollector.Push(m_TensorHandle.get(), m_TensorInfo);