diff --git a/matlab/initial_estimation_checks.m b/matlab/initial_estimation_checks.m
index 8259c7931..0d6463f19 100644
--- a/matlab/initial_estimation_checks.m
+++ b/matlab/initial_estimation_checks.m
@@ -40,7 +40,7 @@ end
 % Evaluate the likelihood.
 [fval,junk1,junk2,a,b,c,d] = feval(objective_function,xparam1,DynareDataset,DynareOptions,Model,EstimatedParameters,BayesInfo,DynareResults);
 
-if DynareOptions.dsge_var
+if DynareOptions.dsge_var || strcmp(func2str(objective_function),'non_linear_dsge_likelihood')
     info = b;
 else
     info = d;