1 files changed, 4 insertions, 2 deletions
diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.hpp b/tests/ImageTensorGenerator/ImageTensorGenerator.hpp
index 3e164bc9e0..b1cb5e36f5 100644
--- a/tests/ImageTensorGenerator/ImageTensorGenerator.hpp
+++ b/tests/ImageTensorGenerator/ImageTensorGenerator.hpp
@@ -4,9 +4,11 @@
 //
 
 #include "../InferenceTestImage.hpp"
-#include "Permute.hpp"
+
 #include <armnn/TypesUtils.hpp>
 
+#include <armnnUtils/Permute.hpp>
+
 #include <algorithm>
 #include <fstream>
 #include <iterator>
@@ -178,4 +180,4 @@ template <>
 void WriteImageTensorImpl<uint8_t>(const std::vector<uint8_t>& imageData, std::ofstream& imageTensorFile)
 {
     std::copy(imageData.begin(), imageData.end(), std::ostream_iterator<int>(imageTensorFile, " "));
-}
-\ No newline at end of file
+}