diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index ed94eba..0622516 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -3,6 +3,7 @@ include_directories(
     ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/lib/json/include
     ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src
     ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/classifiers
+    ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/ensembles
     ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/bayesian_network
     ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/utils
     ${CMAKE_BINARY_DIR}/configured_files/include
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 465f026..9671bb4 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -7,6 +7,7 @@ if(ENABLE_TESTING)
         ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/lib/mdlp
         ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src
         ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/classifiers
+        ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/ensembles
         ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/bayesian_network
         ${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/utils
         ${Python3_INCLUDE_DIRS}