diff --git a/backend/categorisation/processor.py b/backend/categorisation/processor.py index fd0d9c89..ca58fb9d 100644 --- a/backend/categorisation/processor.py +++ b/backend/categorisation/processor.py @@ -79,9 +79,9 @@ def choose_cheapest_relevant_plan( if (plan.record.cost_of_works or float("inf")) == min_cost ] - for priority_plan_id in scenario_priority_order: + for priority_scenario_id in scenario_priority_order: for plan in cheapest_plans: - if plan.id == priority_plan_id: + if plan.scenario.id == priority_scenario_id: return plan return cheapest_plans[0]