From a581788bffa22cfbe307a0471aa9eb31308fc01b Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Sun, 16 Jul 2023 22:04:48 -0400
Subject: [PATCH] simplify return

---
 gtsam/hybrid/HybridGaussianFactorGraph.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/gtsam/hybrid/HybridGaussianFactorGraph.cpp b/gtsam/hybrid/HybridGaussianFactorGraph.cpp
index 10b8de797..dc11cedb0 100644
--- a/gtsam/hybrid/HybridGaussianFactorGraph.cpp
+++ b/gtsam/hybrid/HybridGaussianFactorGraph.cpp
@@ -272,9 +272,9 @@ hybridElimination(const HybridGaussianFactorGraph &factors,
     };
 
     DecisionTree<Key, double> probabilities(eliminationResults, probability);
-    return {
-        std::make_shared<HybridConditional>(gaussianMixture),
-        std::make_shared<TableFactor>(discreteSeparator, dtf->probabilities())};
+
+    return {std::make_shared<HybridConditional>(gaussianMixture),
+            std::make_shared<TableFactor>(discreteSeparator, probabilities)};
   } else {
     // Otherwise, we create a resulting GaussianMixtureFactor on the separator,
     // taking care to correct for conditional constant.