1 files changed, 2 insertions, 3 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index d4a0a6e865..8941ee93f5 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -1963,10 +1963,9 @@ void TfLiteParserImpl::ParsePrelu(size_t subgraphIndex, size_t operatorIndex)
 
     if (IsConstTensor(inputs[1]))
     {
-        armnn::IInputSlot* slot = &(layer->GetInputSlot(0));
-        RegisterConsumerOfTensor(subgraphIndex, 0, slot);
-
         auto inputTensorIndexes = AsUnsignedVector(GetInputTensorIds(m_Model, subgraphIndex, operatorIndex));
+        armnn::IInputSlot* slot = &(layer->GetInputSlot(0));
+        RegisterConsumerOfTensor(subgraphIndex, inputTensorIndexes[0], slot);
 
         auto alphaTensorAndData = CreateConstTensorNonPermuted(inputs[1], alphaTensorInfo);
         std::string constLayerName = fmt::format("Constant:{}", inputs[1]->name);