diff --git a/src/ComputingTasks.cc b/src/ComputingTasks.cc
index 5653b4931ba3e3f8df989b6ae6b5b51c33fb2243..5eac1f8422558aba4974ab655a06441e2e39505c 100644
--- a/src/ComputingTasks.cc
+++ b/src/ComputingTasks.cc
@@ -658,7 +658,7 @@ StochSimulStatement::writeOutput(ostream &output, const string &basename, bool m
 
   options_list.writeOutput(output);
   symbol_list.writeOutput("var_list_", output);
-  output << "info = stoch_simul(var_list_);" << endl;
+  output << "[info, oo_, options_] = stoch_simul(M_, options_, oo_, var_list_);" << endl;
 }
 
 void
@@ -1076,7 +1076,7 @@ DiscretionaryPolicyStatement::writeOutput(ostream &output, const string &basenam
 
   options_list.writeOutput(output);
   symbol_list.writeOutput("var_list_", output);
-  output << "discretionary_policy(var_list_);" << endl;
+  output << "[info, oo_, options_] = discretionary_policy(M_, options_, oo_, var_list_);" << endl;
 }
 
 void