1 files changed, 2 insertions, 3 deletions
diff --git a/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.cpp b/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.cpp
index ba353a32c7..8b08f01b23 100644
--- a/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.cpp
+++ b/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.cpp
@@ -20,13 +20,12 @@ TosaRefPreCompiledWorkload::TosaRefPreCompiledWorkload(const PreCompiledQueueDes
                 "TosaRefPreCompiledWorkload requires a valid pre-compiled object (TosaSerializationHandler).");
     }
 }
-
 void TosaRefPreCompiledWorkload::Execute() const
 {
     tosa::TosaSerializationHandler* handler = static_cast<tosa::TosaSerializationHandler*>(m_Data.m_PreCompiledObject);
 
-    std::vector<std::string> inputNames = handler->GetInputs();
-    std::vector<std::string> outputNames = handler->GetOutputs();
+    std::vector<std::string> inputNames = handler->GetMainRegion()->GetBlocks()[0]->GetInputs();
+    std::vector<std::string> outputNames = handler->GetMainRegion()->GetBlocks()[0]->GetOutputs();
 
     TosaReference::IModelRunner runner;
     GraphStatus status;