diff --git a/lib/services/rocket.py b/lib/services/rocket.py index c5d6554..572d7a2 100644 --- a/lib/services/rocket.py +++ b/lib/services/rocket.py @@ -45,12 +45,8 @@ def from_rocket_model(cls, rocket: Rocket) -> Self: radius=rocket.radius, mass=rocket.mass, inertia=rocket.inertia, - power_off_drag=( - rocket.power_off_drag if rocket.power_off_drag else None - ), - power_on_drag=( - rocket.power_on_drag if rocket.power_on_drag else None - ), + power_off_drag=(None or rocket.power_off_drag), + power_on_drag=(None or rocket.power_on_drag), center_of_mass_without_motor=rocket.center_of_mass_without_motor, coordinate_system_orientation=rocket.coordinate_system_orientation.value.lower(), )