diff --git a/src/mlpack/methods/decision_tree/CMakeLists.txt b/src/mlpack/methods/decision_tree/CMakeLists.txt
index 4072a5097d..d36b3d3681 100644
--- a/src/mlpack/methods/decision_tree/CMakeLists.txt
+++ b/src/mlpack/methods/decision_tree/CMakeLists.txt
@@ -4,6 +4,8 @@ set(SOURCES
   all_dimension_select.hpp
   decision_tree.hpp
   decision_tree_impl.hpp
+  decision_tree_regressor.hpp
+  decision_tree_regressor_impl.hpp
   all_categorical_split.hpp
   all_categorical_split_impl.hpp
   best_binary_numeric_split.hpp