diff --git a/internal/ceres/generate_bundle_adjustment_tests.py b/internal/ceres/generate_bundle_adjustment_tests.py
index 0359b40..d8d1646 100644
--- a/internal/ceres/generate_bundle_adjustment_tests.py
+++ b/internal/ceres/generate_bundle_adjustment_tests.py
@@ -140,7 +140,11 @@
 
 #include "ceres/bundle_adjustment_test_util.h"
 #include "ceres/internal/config.h"
+#include "ceres/problem.h"
+#include "ceres/solver.h"
+#include "ceres/types.h"
 #include "gtest/gtest.h"
+
 %(preprocessor_conditions_begin)s
 namespace ceres::internal {