diff --git a/CMakeLists.txt b/CMakeLists.txt
index b2206b0..14db1ab 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -80,7 +80,7 @@ corrosion_set_env_vars(
     "DUCKDB_VERSION=${DUCKDB_VERSION}"
 )
 
-include_directories(src/include vortex/vortex-duckdb/include)
+include_directories(src/include vortex/vortex-duckdb/include vortex/vortex-duckdb/cpp/include)
 
 set(EXTENSION_NAME ${TARGET_NAME}_extension)
 set(LOADABLE_EXTENSION_NAME ${TARGET_NAME}_loadable_extension)
diff --git a/src/vortex_extension.cpp b/src/vortex_extension.cpp
index 3119973..d48651e 100644
--- a/src/vortex_extension.cpp
+++ b/src/vortex_extension.cpp
@@ -1,6 +1,7 @@
 #define DUCKDB_EXTENSION_MAIN
 
 #include "vortex_extension.hpp"
+#include "duckdb_vx/table_function.h"
 #include "vortex.h"
 
 using namespace duckdb;