diff --git a/Android.mk b/Android.mk
index 2f5b977..2e83e27 100644
--- a/Android.mk
+++ b/Android.mk
@@ -53,9 +53,9 @@
 # ARMNN_COMPUTE_NEON_ENABLED is undefined, use the following default value
 ARMNN_COMPUTE_NEON_ENABLED := 1
 endif
-ifndef ARMNN_COMPUTE_REF_ENABLED
-# ARMNN_COMPUTE_REF_ENABLED is undefined, use the following default value
-ARMNN_COMPUTE_REF_ENABLED := 1
+ifndef ARMNN_REF_ENABLED
+# ARMNN_REF_ENABLED is undefined, use the following default value
+ARMNN_REF_ENABLED := 1
 endif
 
 #######################
@@ -114,10 +114,10 @@
         -DARMCOMPUTENEON_ENABLED
 endif # ARMNN_COMPUTE_NEON_ENABLED == 1
 
-ifeq ($(ARMNN_COMPUTE_REF_ENABLED),1)
+ifeq ($(ARMNN_REF_ENABLED),1)
 LOCAL_CFLAGS += \
         -DARMNNREF_ENABLED
-endif # ARMNN_COMPUTE_REF_ENABLED == 1
+endif # ARMNN_REF_ENABLED == 1
 
 LOCAL_SRC_FILES := \
         1.0/ArmnnDriverImpl.cpp \
@@ -231,10 +231,10 @@
         -DARMCOMPUTENEON_ENABLED
 endif # ARMNN_COMPUTE_NEON_ENABLED == 1
 
-ifeq ($(ARMNN_COMPUTE_REF_ENABLED),1)
+ifeq ($(ARMNN_REF_ENABLED),1)
 LOCAL_CFLAGS += \
         -DARMNNREF_ENABLED
-endif # ARMNN_COMPUTE_REF_ENABLED == 1
+endif # ARMNN_REF_ENABLED == 1
 
 LOCAL_SRC_FILES := \
         1.0/ArmnnDriverImpl.cpp \
@@ -340,10 +340,10 @@
         -DARMCOMPUTENEON_ENABLED
 endif # ARMNN_COMPUTE_NEON_ENABLED == 1
 
-ifeq ($(ARMNN_COMPUTE_REF_ENABLED),1)
+ifeq ($(ARMNN_REF_ENABLED),1)
 LOCAL_CFLAGS += \
         -DARMNNREF_ENABLED
-endif # ARMNN_COMPUTE_REF_ENABLED == 1
+endif # ARMNN_REF_ENABLED == 1
 
 LOCAL_SRC_FILES := \
         1.0/ArmnnDriverImpl.cpp \