diff --git a/main.py b/main.py
index e361797..539875b 100644
--- a/main.py
+++ b/main.py
@@ -275,7 +275,11 @@ class BeamlineEditor(QtWidgets.QMainWindow, Ui_BeamlineGUI):
                 self.elementDB[ele].Prefix=name
                 if ele in self.branchingID.keys():
                     if self.branchingID[ele] is True:
+                        self.elementDB[ele].angle=self.elementDB[ele].design_angle
+                        line.LengthRes=0  # needs to shorten the line container
                         return True
+                    else:
+                        self.elementDB[ele].angle=0
         if secondBranch:
             if tree.secondary:
                 self.unwrap(name, tree.secondary.node_id, line, case_num)