diff --git a/ecp5/arch.cc b/ecp5/arch.cc
index 79f65562..b070014e 100644
--- a/ecp5/arch.cc
+++ b/ecp5/arch.cc
@@ -413,7 +413,7 @@ delay_t Arch::estimateDelay(WireId src, WireId dst) const
     return 200 * (abs(src.location.x - dst.location.x) + abs(src.location.y - dst.location.y));
 }
 
-delay_t Arch::getBudgetOverride(const PortRef &pr, delay_t v) const { return v; }
+delay_t Arch::getBudgetOverride(NetInfo *net_info, int user_idx, delay_t budget) const { return budget; }
 
 // -----------------------------------------------------------------------
 
diff --git a/ecp5/arch.h b/ecp5/arch.h
index 860b086d..445f0dbf 100644
--- a/ecp5/arch.h
+++ b/ecp5/arch.h
@@ -780,7 +780,7 @@ struct Arch : BaseCtx
     delay_t getRipupDelayPenalty() const { return 200; }
     float getDelayNS(delay_t v) const { return v * 0.001; }
     uint32_t getDelayChecksum(delay_t v) const { return v; }
-    delay_t getBudgetOverride(const PortRef &pr, delay_t v) const;
+    delay_t getBudgetOverride(NetInfo *net_info, int user_idx, delay_t budget) const;
 
     // -------------------------------------------------
 
diff --git a/generic/arch.cc b/generic/arch.cc
index 03a6b122..ce6fe50a 100644
--- a/generic/arch.cc
+++ b/generic/arch.cc
@@ -403,7 +403,7 @@ delay_t Arch::estimateDelay(WireId src, WireId dst) const
     return (dx + dy) * grid_distance_to_delay;
 }
 
-delay_t Arch::getBudgetOverride(const PortRef &pr, delay_t v) const { return v; }
+delay_t Arch::getBudgetOverride(NetInfo *net_info, int user_idx, delay_t budget) const { return budget; }
 
 // ---------------------------------------------------------------
 
diff --git a/generic/arch.h b/generic/arch.h
index df21fbe4..b6892c85 100644
--- a/generic/arch.h
+++ b/generic/arch.h
@@ -198,7 +198,7 @@ struct Arch : BaseCtx
     delay_t getRipupDelayPenalty() const { return 1.0; }
     float getDelayNS(delay_t v) const { return v; }
     uint32_t getDelayChecksum(delay_t v) const { return 0; }
-    delay_t getBudgetOverride(const PortRef &pr, delay_t v) const;
+    delay_t getBudgetOverride(NetInfo *net_info, int user_idx, delay_t budget) const;
 
     bool pack() { return true; }
     bool place();