diff --git a/level_zero/tools/source/sysman/memory/linux/CMakeLists.txt b/level_zero/tools/source/sysman/memory/linux/CMakeLists.txt
index 5e1ecaf96b..fead9030ea 100755
--- a/level_zero/tools/source/sysman/memory/linux/CMakeLists.txt
+++ b/level_zero/tools/source/sysman/memory/linux/CMakeLists.txt
@@ -7,7 +7,7 @@
 if(SUPPORT_DG1)
   set(L0_SRCS_TOOLS_SYSMAN_MEMORY_LINUX
       ${CMAKE_CURRENT_SOURCE_DIR}/CMakeLists.txt
-      ${CMAKE_CURRENT_SOURCE_DIR}/dg1/os_memory_imp.cpp
+      ${CMAKE_CURRENT_SOURCE_DIR}/dg1${BRANCH_DIR_SUFFIX}/os_memory_imp.cpp
       ${CMAKE_CURRENT_SOURCE_DIR}/dg1/os_memory_imp.h
   )
 else()
diff --git a/level_zero/tools/test/unit_tests/sources/sysman/memory/linux/CMakeLists.txt b/level_zero/tools/test/unit_tests/sources/sysman/memory/linux/CMakeLists.txt
index 31d94c3087..a36db8e500 100644
--- a/level_zero/tools/test/unit_tests/sources/sysman/memory/linux/CMakeLists.txt
+++ b/level_zero/tools/test/unit_tests/sources/sysman/memory/linux/CMakeLists.txt
@@ -8,7 +8,7 @@ if(SUPPORT_DG1)
   set(L0_TESTS_TOOLS_SYSMAN_MEMORY_LINUX
       ${CMAKE_CURRENT_SOURCE_DIR}/CMakeLists.txt
       ${CMAKE_CURRENT_SOURCE_DIR}/dg1/mock_memory.h
-      ${CMAKE_CURRENT_SOURCE_DIR}/dg1/test_sysman_memory.cpp
+      ${CMAKE_CURRENT_SOURCE_DIR}/dg1${BRANCH_DIR_SUFFIX}/test_sysman_memory.cpp
   )
 else()
   set(L0_TESTS_TOOLS_SYSMAN_MEMORY_LINUX