diff --git a/+nosnoc/+discrete_time_problem/Gcs.m b/+nosnoc/+discrete_time_problem/Gcs.m
index c95f984a..5cf33da2 100644
--- a/+nosnoc/+discrete_time_problem/Gcs.m
+++ b/+nosnoc/+discrete_time_problem/Gcs.m
@@ -95,7 +95,7 @@ function create_variables(obj)
                     obj.w.v(ii,1:opts.N_finite_elements(ii),1:opts.n_s) = {{'v', dims.n_x}};
                 end
                 
-                obj.w.z(ii,1:opts.N_finite_elements(ii),(opts.n_s+rbp)) = {{'z', dims.n_z}, model.lbz, model.ubz, model.z0};
+                obj.w.z(ii,1:opts.N_finite_elements(ii),1:(opts.n_s+rbp)) = {{'z', dims.n_z}, model.lbz, model.ubz, model.z0};
                 obj.w.lambda(ii,1:opts.N_finite_elements(ii),1:(opts.n_s+rbp)) = {{'lambda', dims.n_lambda},0, inf};
                 
                 % Handle x_box settings