diff --git a/matlab/DsgeSmoother.m b/matlab/DsgeSmoother.m
index 8a4ba7e65..3c62224a2 100644
--- a/matlab/DsgeSmoother.m
+++ b/matlab/DsgeSmoother.m
@@ -269,10 +269,10 @@ function [alphahat,etahat,epsilonhat,ahat,SteadyState,trend_coeff,aK,T,R,P,PK,d,
       end
       if kalman_algo == 2
           if missing_value
-              [alphahat,etahat,ahat,aK] = missing_DiffuseKalmanSmoother3(T,R,Q, ...
+              [alphahat,etahat,ahat,P,aK,PK,d,decomp] = missing_DiffuseKalmanSmoother3(T,R,Q, ...
                                                   Pinf,Pstar,Y,trend,nobs,np,smpl,mf,data_index);
           else
-              [alphahat,etahat,ahat,aK] = DiffuseKalmanSmoother3(T,R,Q, ...
+              [alphahat,etahat,ahat,P,aK,PK,d,decomp] = DiffuseKalmanSmoother3(T,R,Q, ...
                                                   Pinf,Pstar,Y,trend,nobs,np,smpl,mf);
           end
       end