--- a/SFCViewer.py	Thu Feb 01 18:09:34 2007 +0100
+++ b/SFCViewer.py	Wed Feb 07 18:43:32 2007 +0100
@@ -75,7 +75,7 @@
             wire.SetPoints([wxPoint(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wxPoint(pos.x, pos.y)])
             if isinstance(next_block, SFC_Divergence):
                 next_block.RefreshPosition()
-            next_block.RefreshModel()
+            transition.RefreshOutputModel(True)
         return transition
     
     def RemoveTransition(self, transition):
@@ -133,7 +133,7 @@
             wire.SetPoints([wxPoint(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wxPoint(pos.x, pos.y)])
             if isinstance(next_block, SFC_Divergence):
                 next_block.RefreshPosition()
-            next_block.RefreshModel()
+            step.RefreshOutputModel(True)
         return step
 
     def RemoveStep(self, step):
@@ -795,7 +795,7 @@
                     wire.SetPoints([wxPoint(pos.x, pos.y + wire_size), wxPoint(pos.x, pos.y)])
                     if isinstance(next_block, SFC_Divergence):
                         next_block.RefreshPosition()
-                    next_block.RefreshModel()
+                    previous_block.RefreshOutputModel(True)
             else:
                 if isinstance(previous_block, SFC_Step):
                     previous_block.RemoveOutput()
@@ -910,7 +910,7 @@
                     wxPoint(previous_pos.x, previous_pos.y)])
                 if isinstance(next_block, SFC_Divergence):
                     next_block.RefreshPosition()
-                next_block.RefreshModel()
+                previous_block.RefreshOutputModel(True)
         elif divergence.GetBranchNumber() == 1:
             wires = connectors["inputs"][0].GetWires()
             if len(wires) != 1:
@@ -943,7 +943,7 @@
                 wxPoint(previous_pos.x, previous_pos.y)])
             if isinstance(next_block, SFC_Divergence):
                 next_block.RefreshPosition()
-            next_block.RefreshModel()
+            previous_block.RefreshOutputModel(True)
         self.Parent.RefreshProjectTree()
 
     def DeleteJump(self, jump):