diff --git a/src/backends/tosaCommon/TosaMappings.cpp b/src/backends/tosaCommon/TosaMappings.cpp
index 3932b62..6ab1b06 100644
--- a/src/backends/tosaCommon/TosaMappings.cpp
+++ b/src/backends/tosaCommon/TosaMappings.cpp
@@ -101,7 +101,7 @@
     std::vector<const TensorInfo*> inputs;
     for (auto inputSlot : layer->GetInputSlots())
     {
-        inputs.push_back(&inputSlot.GetConnection()->GetTensorInfo());
+        inputs.push_back(&inputSlot.GetTensorInfo());
     }
 
     std::vector<const TensorInfo*> outputs;