about summary refs log tree commit diff
path: root/stage.py
diff options
context:
space:
mode:
Diffstat (limited to 'stage.py')
-rw-r--r--stage.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/stage.py b/stage.py
index 783e1f4..0df1d52 100644
--- a/stage.py
+++ b/stage.py
@@ -18,9 +18,9 @@ class Stage():
     def total_mass(self):
         return (self.stage_mass + self.fuel_mass)
 
-    def current_thrust(self):
+    def current_thrust(self, g: float):
         if(self.engines_on and self.fuel_mass > 0):
-            return self.engine.thrust(self.throttle) * self.engine_number
+            return self.engine.thrust(self.throttle, g) * self.engine_number
         else:
             return 0