diff --git a/tests/SConscript b/tests/SConscript
index 291a7a5..77d370f 100644
--- a/tests/SConscript
+++ b/tests/SConscript
@@ -150,7 +150,7 @@
 if env['opencl']:
     Import('opencl')
 
-    benchmark_env.Append(CPPDEFINES=['OPENCL'])
+    benchmark_env.Append(CPPDEFINES=['ARM_COMPUTE_CL'])
 
     files_benchmark += Glob('new/benchmark_new/CL/*/*.cpp')
     files_benchmark += Glob('new/benchmark_new/CL/*.cpp')
diff --git a/tests/benchmark_new/main.cpp b/tests/benchmark_new/main.cpp
index 4c6811e..46c51b6 100644
--- a/tests/benchmark_new/main.cpp
+++ b/tests/benchmark_new/main.cpp
@@ -30,9 +30,9 @@
 #include "support/ToolchainSupport.h"
 #include "tests/TensorLibrary.h"
 
-#ifdef OPENCL
+#ifdef ARM_COMPUTE_CL
 #include "arm_compute/runtime/CL/CLScheduler.h"
-#endif /* OPENCL */
+#endif /* ARM_COMPUTE_CL */
 #include "arm_compute/runtime/Scheduler.h"
 
 #include <fstream>
@@ -55,9 +55,9 @@
 
 int main(int argc, char **argv)
 {
-#ifdef OPENCL
+#ifdef ARM_COMPUTE_CL
     CLScheduler::get().default_init();
-#endif /* OPENCL */
+#endif /* ARM_COMPUTE_CL */
 
     framework::Framework &framework = framework::Framework::get();