Fix stop command (#1841)

2 files changed, 6 insertions, 3 deletions

diff --git a/lib/stitch_plan/stitch_plan.py b/lib/stitch_plan/stitch_plan.py
index 04d587c2..313bc6e0 100644
--- a/lib/stitch_plan/stitch_plan.py
+++ b/lib/stitch_plan/stitch_plan.py
@@ -40,6 +40,9 @@ def stitch_groups_to_stitch_plan(stitch_groups, collapse_len=None, min_stitch_le
         if color_block.color != stitch_group.color:
             if len(color_block) == 0:
                 # We just processed a stop, which created a new color block.
+                # end the previous block with a color change
+                stitch_plan.color_blocks[-2].add_stitch(color_change=True)
+
                 # We'll just claim this new block as ours:
                 color_block.color = stitch_group.color
             else:
@@ -49,8 +52,8 @@ def stitch_groups_to_stitch_plan(stitch_groups, collapse_len=None, min_stitch_le
                 # make a new block of our color
                 color_block = stitch_plan.new_color_block(color=stitch_group.color)
 
-                # always start a color with a JUMP to the first stitch position
-                color_block.add_stitch(stitch_group.stitches[0], jump=True, tie_modus=stitch_group.tie_modus)
+            # always start a color with a JUMP to the first stitch position
+            color_block.add_stitch(stitch_group.stitches[0], jump=True, tie_modus=stitch_group.tie_modus)
         else:
             if (len(color_block) and
                     ((stitch_group.stitches[0] - color_block.stitches[-1]).length() > collapse_len or
diff --git a/lib/threads/color.py b/lib/threads/color.py
index 4a2f58e6..85c8948c 100644
--- a/lib/threads/color.py
+++ b/lib/threads/color.py
@@ -66,7 +66,7 @@ class ThreadColor(object):
         return hash(self.rgb)
 
     def __ne__(self, other):
-        return not(self == other)
+        return not (self == other)
 
     def __repr__(self):
         return "ThreadColor" + repr(self.rgb)