diff --git a/CMakeLists.txt b/CMakeLists.txt
index 780b9a5fc82a1bd5cd95f736c1b7a3db40364589..89cdb4513c77788cf1a3cc63ff5e80590a60381a 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -553,11 +553,13 @@ endif ( )
 #############################################################################################################################
 if ( WALBERLA_BUILD_WITH_CODEGEN )
     find_package( PythonInterp 3 QUIET REQUIRED)
-    execute_process(COMMAND ${PYTHON_EXECUTABLE} -c "import pystencils_walberla" RESULT_VARIABLE PYTHON_RET_CODE)
+    execute_process(COMMAND ${PYTHON_EXECUTABLE} -c "import pystencils_walberla; from pystencils.include import get_pystencils_include_path; print(get_pystencils_include_path())"
+                    RESULT_VARIABLE PYTHON_RET_CODE
+                    OUTPUT_VARIABLE PYSTENCILS_INCLUDE_PATH)
     if(NOT PYTHON_RET_CODE EQUAL 0)
         message(FATAL_ERROR "WALBERLA_BUILD_WITH_CODEGEN activated and pystencils_walberla package not found")
     endif()
-
+    include_directories( ${PYSTENCILS_INCLUDE_PATH} )
 endif()
 ############################################################################################################################