diff --git a/CMakeLists.txt b/CMakeLists.txt
index e8f9c46..dfd10be 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -394,6 +394,8 @@
         src/armnn/test/TensorTest.cpp
         src/armnn/test/TestLayerVisitor.cpp
         src/armnn/test/TestLayerVisitor.hpp
+        src/armnn/test/TestNameOnlyLayerVisitor.cpp
+        src/armnn/test/TestNameOnlyLayerVisitor.hpp
         src/armnn/test/UnitTests.cpp
         src/armnn/test/UnitTests.hpp
         src/armnn/test/UtilsTests.cpp