1 files changed, 1 insertions, 3 deletions
diff --git a/tests/TfLiteMobilenetV2Quantized-Armnn/TfLiteMobilenetV2Quantized-Armnn.cpp b/tests/TfLiteMobilenetV2Quantized-Armnn/TfLiteMobilenetV2Quantized-Armnn.cpp
index 5db5c243b1..9bc1034fea 100644
--- a/tests/TfLiteMobilenetV2Quantized-Armnn/TfLiteMobilenetV2Quantized-Armnn.cpp
+++ b/tests/TfLiteMobilenetV2Quantized-Armnn/TfLiteMobilenetV2Quantized-Armnn.cpp
@@ -40,14 +40,12 @@ int main(int argc, char* argv[])
                      [&imageSet](const char* dataDir, const ModelType & model) {
                          // we need to get the input quantization parameters from
                          // the parsed model
-                         auto inputBinding = model.GetInputBindingInfo();
                          return DatabaseType(
                              dataDir,
                              224,
                              224,
                              imageSet,
-                             inputBinding.second.GetQuantizationScale(),
-                             inputBinding.second.GetQuantizationOffset());
+                             1);
                      },
                      &inputTensorShape);
     }