State feedback loop now supports gain scheduling.
diff --git a/frc971/control_loops/index/index.cc b/frc971/control_loops/index/index.cc
index 9ecc1da..4053eae 100644
--- a/frc971/control_loops/index/index.cc
+++ b/frc971/control_loops/index/index.cc
@@ -217,10 +217,10 @@
   }
 
   for (int i = 0; i < kNumOutputs; ++i) {
-    if (U[i] > plant.U_max[i]) {
-      U[i] = plant.U_max[i];
-    } else if (U[i] < plant.U_min[i]) {
-      U[i] = plant.U_min[i];
+    if (U(i, 0) > U_max(i, 0)) {
+      U(i, 0) = U_max(i, 0);
+    } else if (U(i, 0) < U_min(i, 0)) {
+      U(i, 0) = U_min(i, 0);
     }
   }
 }