diff --git a/src/teakra/src/CMakeLists.txt b/src/teakra/src/CMakeLists.txt
index b96c500b..30683374 100644
--- a/src/teakra/src/CMakeLists.txt
+++ b/src/teakra/src/CMakeLists.txt
@@ -32,11 +32,16 @@ add_library(teakra
     register.h
     shared_memory.h
     teakra.cpp
-    test.h
-    test_generator.cpp
-    test_generator.h
 )
 
+if (TEAKRA_BUILD_UNIT_TESTS)
+    target_sources(teakra PUBLIC
+        test.h
+        test_generator.cpp
+        test_generator.h
+    )
+endif()
+
 create_target_directory_groups(teakra)
 
 target_link_libraries(teakra PRIVATE Threads::Threads)