diff --git a/gtsam/hybrid/tests/testHybridEstimation.cpp b/gtsam/hybrid/tests/testHybridEstimation.cpp
index 1bcd3ad27..492621228 100644
--- a/gtsam/hybrid/tests/testHybridEstimation.cpp
+++ b/gtsam/hybrid/tests/testHybridEstimation.cpp
@@ -290,7 +290,7 @@ TEST(HybridEstimation, Probability) {
       graph.continuousDelta(discrete_keys, bayesNet, assignments);
 
   AlgebraicDecisionTree<Key> probPrimeTree =
-      graph.continuousProbPrimes(discrete_keys, bayesNet, assignments);
+      graph.continuousProbPrimes(discrete_keys, bayesNet);
 
   EXPECT(assert_equal(expected_probPrimeTree, probPrimeTree));