diff --git a/yaml-reader/CMakeLists.txt b/yaml-reader/CMakeLists.txt
index 866bfe0..f2c3ce2 100644
--- a/yaml-reader/CMakeLists.txt
+++ b/yaml-reader/CMakeLists.txt
@@ -1,6 +1,10 @@
-set(yaml-reader_sources main.cpp tests.cpp)
+file(GLOB yaml-reader_headers *.h)
+file(GLOB yaml-reader_sources *.cpp)
 
-add_executable(yaml-reader ${yaml-reader_sources})
+add_executable(yaml-reader
+	${yaml-reader_sources}
+	${yaml-reader_headers}
+)
 target_link_libraries(yaml-reader yaml-cpp)
 
 add_test(yaml-reader-test yaml-reader)