diff --git a/CMakeLists.txt b/CMakeLists.txt
index 0657338..abd9b63 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -43,10 +43,11 @@ if (NOT DEFINED DSL_MODULE_DIR)
 endif ()
 get_filename_component(DSL_MODULE_DIR ${DSL_MODULE_DIR} REALPATH)
 message(STATUS "DSL module dir: ${DSL_MODULE_DIR}")
-set(DSL_SOURCES ${DSL_MODULE_DIR}/*)
-set(DSL_HEADERS "${PROJECT_BINARY_DIR}/stencil_assembly.cuh"
-                "${PROJECT_BINARY_DIR}/stencil_process.cuh"
-                "${PROJECT_BINARY_DIR}/stencil_defines.h")
+set(DSL_SOURCES "${DSL_MODULE_DIR}/*"
+                "acc/src/*"
+                "${PROJECT_BINARY_DIR}/acc")
+set(DSL_HEADERS "${PROJECT_BINARY_DIR}/user_kernels.h"
+                "${PROJECT_BINARY_DIR}/user_defines.h")
 add_custom_command (
     COMMENT "Building ACC objects ${DSL_MODULE_DIR}"
     COMMAND ${CMAKE_SOURCE_DIR}/scripts/compile_acc_module.sh ${DSL_MODULE_DIR}