diff --git a/api/endpoints/editsalamander.py b/api/endpoints/editsalamander.py index ab2fdc8bea9b4edcc334d577ecc418942534fb6b..2b0d8513151dcee53576438e62223f2fc871b806 100644 --- a/api/endpoints/editsalamander.py +++ b/api/endpoints/editsalamander.py @@ -171,12 +171,13 @@ class EditSalamander(Resource): path_to_original_image=original_path, path_to_processed_image=processed_path) else: - add_salamander(salamander.id, image_id, data['new_location'], - data['new_species'], - data['new_sex'], new_path_to_images, user_id, original_path, - processed_path) + new_id = add_salamander(salamander.id, image_id, data['new_location'], + data['new_species'], + data['new_sex'], new_path_to_images, user_id, + original_path, + processed_path) return handle_remaining_images(salamander, salamander_path, image_id, last_id, - True, result, 0) + True, result, new_id) else: return jsonify( {"message": "Salamander image, " + data['image_id'] + " does not exist",