diff --git a/python/src/main.cpp b/python/src/main.cpp
index 1a20cc9..5bff2d6 100644
--- a/python/src/main.cpp
+++ b/python/src/main.cpp
@@ -2,7 +2,7 @@
 #include <pybind11/stl.h>
 #include <pybind11/numpy.h>
 
-#include <AggregateHeaders.cpp>
+#include <kompute/Kompute.hpp>
 
 #include "docstrings.hpp"