diff --git a/gtsam/discrete/DiscreteConditional.cpp b/gtsam/discrete/DiscreteConditional.cpp
index 2f900afbe..048f35e5b 100644
--- a/gtsam/discrete/DiscreteConditional.cpp
+++ b/gtsam/discrete/DiscreteConditional.cpp
@@ -201,7 +201,7 @@ DiscreteConditional::shared_ptr DiscreteConditional::choose(
 }
 
 /* ************************************************************************** */
-DecisionTreeFactor::shared_ptr DiscreteConditional::likelihood(
+DiscreteFactor::shared_ptr DiscreteConditional::likelihood(
     const DiscreteValues& frontalValues) const {
   // Get the big decision tree with all the levels, and then go down the
   // branches based on the value of the frontal variables.
@@ -226,7 +226,7 @@ DecisionTreeFactor::shared_ptr DiscreteConditional::likelihood(
 }
 
 /* ****************************************************************************/
-DecisionTreeFactor::shared_ptr DiscreteConditional::likelihood(
+DiscreteFactor::shared_ptr DiscreteConditional::likelihood(
     size_t frontal) const {
   if (nrFrontals() != 1)
     throw std::invalid_argument(