diff --git a/gtsam_unstable/discrete/CSP.cpp b/gtsam_unstable/discrete/CSP.cpp
index 283c992f1..e204a6779 100644
--- a/gtsam_unstable/discrete/CSP.cpp
+++ b/gtsam_unstable/discrete/CSP.cpp
@@ -6,6 +6,7 @@
  */
 
 #include <gtsam/base/Testable.h>
+#include <gtsam/discrete/DiscreteBayesNet.h>
 #include <gtsam_unstable/discrete/CSP.h>
 #include <gtsam_unstable/discrete/Domain.h>
 
diff --git a/gtsam_unstable/discrete/Scheduler.h b/gtsam_unstable/discrete/Scheduler.h
index 7559cdea6..a97368bb2 100644
--- a/gtsam_unstable/discrete/Scheduler.h
+++ b/gtsam_unstable/discrete/Scheduler.h
@@ -8,6 +8,7 @@
 #pragma once
 
 #include <gtsam_unstable/discrete/CSP.h>
+#include <gtsam/discrete/DiscreteBayesNet.h>
 
 namespace gtsam {