diff --git a/albatross/core/model.h b/albatross/core/model.h
index d78d2870..dc01a5be 100644
--- a/albatross/core/model.h
+++ b/albatross/core/model.h
@@ -97,7 +97,7 @@ class RegressionModel : public ParameterHandlingMixin {
     return predict(features, detail::PredictTypeIdentity<PredictType>());
   }
 
-  template <typename PredictType>
+  template <typename PredictType = JointDistribution>
   PredictType predict(const FeatureType &feature) const {
     std::vector<FeatureType> features = {feature};
     return predict<PredictType>(features);