diff --git a/arm_compute/runtime/NEON/functions/NEIm2Col.h b/arm_compute/runtime/NEON/functions/NEIm2Col.h
index de4780f..e281dce 100644
--- a/arm_compute/runtime/NEON/functions/NEIm2Col.h
+++ b/arm_compute/runtime/NEON/functions/NEIm2Col.h
@@ -24,7 +24,7 @@
 #ifndef __ARM_COMPUTE_NEIM2COL_H__
 #define __ARM_COMPUTE_NEIM2COL_H__
 
-#include "arm_compute/runtime/NEON/INESimpleFunction.h"
+#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h"
 
 #include "arm_compute/core/NEON/kernels/NEIm2ColKernel.h"
 #include "arm_compute/core/Size2D.h"
@@ -78,5 +78,5 @@
     NEIm2ColKernel _kernel;
     unsigned int   _y_dim;
 };
-}
+} // namespace arm_compute
 #endif /* __ARM_COMPUTE_NEIM2COL_H__ */