diff --git a/drone_flightplan/flightplan.py b/drone_flightplan/flightplan.py index 35d832f..3c10b89 100644 --- a/drone_flightplan/flightplan.py +++ b/drone_flightplan/flightplan.py @@ -56,7 +56,9 @@ def add_speed_and_agl(placemark): updated_waypoints = list(map(add_speed_and_agl, waypoints)) # generate wpml file in output_file_path - create_xml(updated_waypoints, "goHome", agl, output_file_path) + output_file = create_xml(updated_waypoints, "goHome", agl, output_file_path) + + return output_file def main():