diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.hpp b/tests/ImageTensorGenerator/ImageTensorGenerator.hpp
index 3e164bc..b1cb5e3 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 @@
 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
+}