diff --git a/arm_compute/runtime/CL/functions/CLFFT2D.h b/arm_compute/runtime/CL/functions/CLFFT2D.h
index a0673ec..d53f1ef 100644
--- a/arm_compute/runtime/CL/functions/CLFFT2D.h
+++ b/arm_compute/runtime/CL/functions/CLFFT2D.h
@@ -26,10 +26,10 @@
 
 #include "arm_compute/runtime/IFunction.h"
 
-#include "arm_compute/runtime/CL/CLMemoryGroup.h"
 #include "arm_compute/runtime/CL/CLTensor.h"
 #include "arm_compute/runtime/CL/functions/CLFFT1D.h"
 #include "arm_compute/runtime/FunctionDescriptors.h"
+#include "arm_compute/runtime/MemoryGroup.h"
 
 namespace arm_compute
 {
@@ -67,10 +67,10 @@
     void run() override;
 
 protected:
-    CLMemoryGroup _memory_group;
-    CLFFT1D       _first_pass_func;
-    CLFFT1D       _second_pass_func;
-    CLTensor      _first_pass_tensor;
+    MemoryGroup _memory_group;
+    CLFFT1D     _first_pass_func;
+    CLFFT1D     _second_pass_func;
+    CLTensor    _first_pass_tensor;
 };
 } // namespace arm_compute
 #endif /*__ARM_COMPUTE_CLFFT2D_H__ */