diff --git a/src/backends/reference/backend.mk b/src/backends/reference/backend.mk
index 13251b7..4bfacd5 100644
--- a/src/backends/reference/backend.mk
+++ b/src/backends/reference/backend.mk
@@ -7,10 +7,10 @@
 # in the Android build and it is picked up by the Android.mk
 # file in the root of ArmNN
 
-# The variable to enable/disable the reference backend (ARMNN_COMPUTE_REF_ENABLED is declared in android-nn-driver/Android.mk)
-ifeq ($(ARMNN_COMPUTE_REF_ENABLED),1)
+# The variable to enable/disable the reference backend (ARMNN_REF_ENABLED is declared in android-nn-driver/Android.mk)
+ifeq ($(ARMNN_REF_ENABLED),1)
 
-# ARMNN_COMPUTE_REF_ENABLED == 1
+# ARMNN_REF_ENABLED == 1
 # Include the source files for the reference backend
 
 BACKEND_SOURCES := \
@@ -86,7 +86,7 @@
         workloads/TransposeConvolution2d.cpp
 else
 
-# ARMNN_COMPUTE_REF_ENABLED == 0
+# ARMNN_REF_ENABLED == 0
 # No source file will be compiled for the reference backend
 
 BACKEND_SOURCES :=
@@ -97,10 +97,10 @@
 # in the Android unit test build (armnn-tests) and it is picked
 # up by the Android.mk file in the root of ArmNN
 
-# The variable to enable/disable the CL backend (ARMNN_COMPUTE_REF_ENABLED is declared in android-nn-driver/Android.mk)
-ifeq ($(ARMNN_COMPUTE_REF_ENABLED),1)
+# The variable to enable/disable the CL backend (ARMNN_REF_ENABLED is declared in android-nn-driver/Android.mk)
+ifeq ($(ARMNN_REF_ENABLED),1)
 
-# ARMNN_COMPUTE_REF_ENABLED == 1
+# ARMNN_REF_ENABLED == 1
 # Include the source files for the CL backend tests
 
 BACKEND_TEST_SOURCES := \
@@ -115,7 +115,7 @@
         test/RefRuntimeTests.cpp
 else
 
-# ARMNN_COMPUTE_REF_ENABLED == 0
+# ARMNN_REF_ENABLED == 0
 # No source file will be compiled for the reference backend tests
 
 BACKEND_TEST_SOURCES :=