diff --git a/gtsam/hybrid/tests/testHybridGaussianConditional.cpp b/gtsam/hybrid/tests/testHybridGaussianConditional.cpp
index 032be5a78..88a4fa485 100644
--- a/gtsam/hybrid/tests/testHybridGaussianConditional.cpp
+++ b/gtsam/hybrid/tests/testHybridGaussianConditional.cpp
@@ -21,6 +21,7 @@
 #include <gtsam/discrete/DecisionTree.h>
 #include <gtsam/discrete/DiscreteKey.h>
 #include <gtsam/discrete/DiscreteValues.h>
+#include <gtsam/hybrid/HybridConditional.h>
 #include <gtsam/hybrid/HybridGaussianConditional.h>
 #include <gtsam/hybrid/HybridGaussianFactor.h>
 #include <gtsam/hybrid/HybridValues.h>
@@ -304,18 +305,6 @@ TEST(HybridGaussianConditional, Prune) {
   }
 }
 
-/* ************************************************************************* */
-
-#include <gtsam/hybrid/HybridConditional.h>
-
-// Helper function to apply discrete values to the tree
-auto choose(auto tree, const DiscreteValues &discreteValues) {
-  for (const auto &[key, value] : discreteValues) {
-    tree = tree.choose(key, value);
-  }
-  return tree;
-}
-
 /* *************************************************************************
  * This test verifies the behavior of the restrict method in different
  * scenarios: