diff --git a/gtsam/discrete/DecisionTreeFactor.h b/gtsam/discrete/DecisionTreeFactor.h
index 3cfdbe968..1f3d69292 100644
--- a/gtsam/discrete/DecisionTreeFactor.h
+++ b/gtsam/discrete/DecisionTreeFactor.h
@@ -178,7 +178,7 @@ namespace gtsam {
      * A leaf is pruned if it is not in the top `maxNrLeaves` values.
      *
      * @param maxNrLeaves The maximum number of leaves to keep.
-     * @return DecisionTreeFactor::shared_ptr
+     * @return DecisionTreeFactor
      */
     DecisionTreeFactor prune(size_t maxNrLeaves) const;