diff --git a/src/devices_models/device_constructors/branch_constructor.jl b/src/devices_models/device_constructors/branch_constructor.jl
index bd54cf800..c1eb96899 100644
--- a/src/devices_models/device_constructors/branch_constructor.jl
+++ b/src/devices_models/device_constructors/branch_constructor.jl
@@ -351,7 +351,7 @@ function construct_device!(
 ) where {T <: PSY.ACBranch}
     devices =
         get_available_components(T, sys, get_attribute(model, "filter_function"))
-    branch_rate_bounds!(container, devices, model, typeof(network_model).parameters[1])
+    branch_rate_bounds!(container, devices, model, get_network_formulation(network_model))
     add_constraint_dual!(container, sys, model)
     return
 end