diff --git a/core/src/allocation_init.jl b/core/src/allocation_init.jl
index cd2dd31aa..63221db60 100644
--- a/core/src/allocation_init.jl
+++ b/core/src/allocation_init.jl
@@ -473,7 +473,6 @@ function add_constraints_capacity!(
     (; graph) = p
     main_network_source_edges = get_main_network_connections(p, allocation_network_id)
     F = problem[:F]
-    edge_ids = graph[].edge_ids[allocation_network_id]
     edge_ids_finite_capacity = Tuple{NodeID, NodeID}[]
     for (edge, c) in capacity.data
         if !isinf(c) && edge ∉ main_network_source_edges
diff --git a/core/src/allocation_optim.jl b/core/src/allocation_optim.jl
index a34a75476..049dfbd7c 100644
--- a/core/src/allocation_optim.jl
+++ b/core/src/allocation_optim.jl
@@ -273,9 +273,7 @@ function adjust_capacities_edge!(
     p::Parameters,
     priority_idx::Int,
 )::Nothing
-    (; graph) = p
     (; problem, capacity, allocation_network_id) = allocation_model
-    edge_ids = graph[].edge_ids[allocation_network_id]
     constraints_capacity = problem[:capacity]
     F = problem[:F]