diff --git a/CMakeLists.txt b/CMakeLists.txt
index 9a9e7ebb95ea73e5f5b1e89dedcc798bfebe7c1b..8ede1a1a4d361a36b57639798d0f40164e4ab340 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -17,10 +17,19 @@ endif(INCLUDE_CUDA)
 set(SOURCES_CXX 
     src/MemoryProcessing.cpp
     )
+set(HEADERS_CXX 
+    include/MemoryProcessing.h
+    )
 if(INCLUDE_CUDA)
     set(SOURCES_CU 
     src/MemoryProcessing.cu
     )
+    set(HEADERS_CU
+    include/MemoryProcessing.cuh
+    )
 endif(INCLUDE_CUDA)
 
-add_library(memproc STATIC ${SOURCES_CU} ${SOURCES_CXX})
\ No newline at end of file
+set(SOURCES ${SOURCES_CU} ${SOURCES_CXX})
+set(HEADERS ${HEADERS_CU} ${HEADERS_CXX})
+
+add_library(memproc STATIC ${HEADERS} ${SOURCES})
\ No newline at end of file