diff --git a/src/MOI_wrapper/MOI_wrapper.jl b/src/MOI_wrapper/MOI_wrapper.jl
index a98cbfc..5c9438c 100644
--- a/src/MOI_wrapper/MOI_wrapper.jl
+++ b/src/MOI_wrapper/MOI_wrapper.jl
@@ -258,7 +258,7 @@ function _optimize!(dest::Optimizer, src::OptimizerCache)
         A.n,
         G.m,
         A.m,
-        Gh.sets.num_rows[1],
+        MOI.Utilities.num_rows(Gh.sets, MOI.Nonnegatives),
         length(q),
         q,
         num_exponential,