diff --git a/nullptr/mission.py b/nullptr/mission.py index a0fbb32..44b2a15 100644 --- a/nullptr/mission.py +++ b/nullptr/mission.py @@ -337,7 +337,8 @@ class HaulMission(BaseMission): 'more': 'travel', 'done': 'dock' }), - 'dock': (self.step_dock, 'load'), + 'dock': (self.step_dock, 'refuel'), + 'refuel': (self.step_unload, 'load'), 'load': (self.step_load, 'orbit'), 'orbit': (self.step_orbit, 'travel_back'), 'travel_back': (self.step_travel_dest, { @@ -345,8 +346,8 @@ class HaulMission(BaseMission): 'done': 'dock2' }), 'dock2': (self.step_dock, 'unload'), - 'unload': (self.step_unload, 'refuel'), - 'refuel': (self.step_refuel, 'orbit2'), + 'unload': (self.step_unload, 'refuel2'), + 'refuel2': (self.step_refuel, 'orbit2'), 'orbit2': (self.step_orbit, 'travel') }