1 files changed, 5 insertions, 4 deletions
diff --git a/src/armnn/layers/ArgMinMaxLayer.cpp b/src/armnn/layers/ArgMinMaxLayer.cpp
index bfd71d519b..35bf272e41 100644
--- a/src/armnn/layers/ArgMinMaxLayer.cpp
+++ b/src/armnn/layers/ArgMinMaxLayer.cpp
@@ -2,13 +2,14 @@
 // Copyright © 2017 Arm Ltd. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
-#include "ArgMinMaxLayer.hpp"
 
+#include "ArgMinMaxLayer.hpp"
 #include "LayerCloneBase.hpp"
 
-#include <TensorUtils.hpp>
-
 #include <armnn/TypesUtils.hpp>
+
+#include <armnnUtils/TensorUtils.hpp>
+
 #include <backendsCommon/WorkloadData.hpp>
 #include <backendsCommon/WorkloadFactory.hpp>
 
@@ -88,4 +89,4 @@ void ArgMinMaxLayer::Accept(ILayerVisitor& visitor) const
     visitor.VisitArgMinMaxLayer(this, GetParameters(), GetName());
 }
 
-} // namespace armnn
-\ No newline at end of file
+} // namespace armnn