diff --git a/src/backends/reference/CMakeLists.txt b/src/backends/reference/CMakeLists.txt
index ff25378..a6c76c1 100644
--- a/src/backends/reference/CMakeLists.txt
+++ b/src/backends/reference/CMakeLists.txt
@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: MIT
 #
 
-if(ARMCOMPUTEREF)
+if(ARMNNREF)
     list(APPEND armnnRefBackend_sources
         RefBackend.cpp
         RefBackend.hpp
diff --git a/src/backends/reference/backend.cmake b/src/backends/reference/backend.cmake
index e88968d..01a4ee1 100644
--- a/src/backends/reference/backend.cmake
+++ b/src/backends/reference/backend.cmake
@@ -6,7 +6,7 @@
 add_subdirectory(${PROJECT_SOURCE_DIR}/src/backends/reference)
 list(APPEND armnnLibraries armnnRefBackend)
 
-if(ARMCOMPUTEREF)
+if(ARMNNREF)
     list(APPEND armnnLibraries armnnRefBackendWorkloads)
     list(APPEND armnnUnitTestLibraries armnnRefBackendUnitTests)
 else()