1 files changed, 2 insertions, 2 deletions
diff --git a/tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp b/tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp
index c152c0f6c2..bec2771d4b 100644
--- a/tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp
+++ b/tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp
@@ -3,7 +3,7 @@
 // SPDX-License-Identifier: MIT
 //
 #include "../InferenceTest.hpp"
-#include "../CaffePreprocessor.hpp"
+#include "../ImagePreprocessor.hpp"
 #include "armnnTfParser/ITfParser.hpp"
 
 int main(int argc, char* argv[])
@@ -21,7 +21,7 @@ int main(int argc, char* argv[])
         armnn::TensorShape inputTensorShape({ 1, 3, 224, 224 });
 
         using DataType = float;
-        using DatabaseType = CaffePreprocessor;
+        using DatabaseType = ImagePreprocessor<DataType>;;
         using ParserType = armnnTfParser::ITfParser;
         using ModelType = InferenceModel<ParserType, DataType>;