1 files changed, 1 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h b/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
index efe959f14e..02d0cc15b2 100644
--- a/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
+++ b/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
@@ -25,7 +25,7 @@
 #define ARM_COMPUTE_NESOFTMAXLAYER_H
 
 #include "arm_compute/runtime/IFunction.h"
-#include "arm_compute/runtime/MemoryGroup.h"
+#include "src/core/helpers/MemoryHelpers.h"
 #include <memory>
 
 namespace arm_compute
@@ -88,7 +88,6 @@ public:
     void run() override;
 
 private:
-    MemoryGroup _memory_group;
     struct Impl;
     std::unique_ptr<Impl> _impl;
 };