diff --git a/y2023/control_loops/python/graph_edit.py b/y2023/control_loops/python/graph_edit.py
index 39cd814..c104808 100644
--- a/y2023/control_loops/python/graph_edit.py
+++ b/y2023/control_loops/python/graph_edit.py
@@ -366,7 +366,8 @@
 
     # Current segment based on which mode the drawing system is in.
     def current_seg(self):
-        if self.prev_segment_pt and self.now_segment_pt:
+        if self.prev_segment_pt is not None and (self.prev_segment_pt.any() and
+                                                 self.now_segment_pt.any()):
             if self.theta_version:
                 return AngleSegment(self.prev_segment_pt, self.now_segment_pt)
             else: