diff --git a/src/sampling/sampling_tools.jl b/src/sampling/sampling_tools.jl
index 0762c4cf5af0d6560963b69cd7542417fbe6c202..a73785f2ba51d38217cf765ae387d6f06d454ee2 100644
--- a/src/sampling/sampling_tools.jl
+++ b/src/sampling/sampling_tools.jl
@@ -60,7 +60,9 @@ end
 
 Return Float64 vectors of the lower and upper bounds of the JuMP constraint refs.
 """
-function get_bound_vectors(lbconref, ubconref)
+function get_bound_vectors(opt_model)
+    lbconref = opt_model[:lbs]
+    ubconref = opt_model[:ubs]
     lbs = zeros(length(lbconref))
     for i in eachindex(lbs)
         lbval = normalized_rhs(lbconref[i])