diff --git a/sensitivityAnalysis/src/sensitivityStochRun.m b/sensitivityAnalysis/src/sensitivityStochRun.m
index ca0cbc1bb3f766613705d85c9021a7e0b19f7ccd..6893fba0af4f9ab9d96e456baf294eb5c723c511 100644
--- a/sensitivityAnalysis/src/sensitivityStochRun.m
+++ b/sensitivityAnalysis/src/sensitivityStochRun.m
@@ -107,7 +107,7 @@ for i = 1:n
     rocket = rocketRef;
     env = envRef;
     uncertanty = struct();
-    updateData(rocket, env, wind, parameters, i);
+    [rocket, env, wind] = updateData(rocket, env, wind, parameters, i);
 
     if ascentFlag
         [ascentSol, ascent(i)] = stdAscent(rocket, env, wind, settings, wrapper);
diff --git a/sensitivityAnalysis/src/stochParameters/updateData.m b/sensitivityAnalysis/src/stochParameters/updateData.m
index 1360c7bc76388373d3dc80b386bc7be7e6b9446d..1fd7a723bf96783d5bd0d33b495a40af0ac8e646 100644
--- a/sensitivityAnalysis/src/stochParameters/updateData.m
+++ b/sensitivityAnalysis/src/stochParameters/updateData.m
@@ -1,4 +1,4 @@
-function updateData(rocket, env, wind, parameters, k)
+function [rocket, env, wind] = updateData(rocket, env, wind, parameters, k)
 arguments
     rocket      Rocket
     env         Environment