diff --git a/y2023/control_loops/python/roll.py b/y2023/control_loops/python/roll.py
index 7e342b2..ce0fac0 100644
--- a/y2023/control_loops/python/roll.py
+++ b/y2023/control_loops/python/roll.py
@@ -39,6 +39,7 @@
         R = numpy.matrix([[numpy.pi / 2.0], [0.0]])
         angular_system.PlotKick(kRoll, R)
         angular_system.PlotMotion(kRoll, R)
+        return
 
     # Write the generated constants out to a file.
     if len(argv) != 5:
diff --git a/y2023/control_loops/python/wrist.py b/y2023/control_loops/python/wrist.py
index 26d3a16..1970c5b 100644
--- a/y2023/control_loops/python/wrist.py
+++ b/y2023/control_loops/python/wrist.py
@@ -36,8 +36,9 @@
 def main(argv):
     if FLAGS.plot:
         R = numpy.matrix([[numpy.pi / 2.0], [0.0]])
-        angular_system.PlotKick(kIntake, R)
-        angular_system.PlotMotion(kIntake, R)
+        angular_system.PlotKick(kWrist, R)
+        angular_system.PlotMotion(kWrist, R)
+        return
 
     # Write the generated constants out to a file.
     if len(argv) != 5: