diff --git a/trainbenchmark-tool-jastadd-specialized-base/src/main/java/de/tudresden/inf/st/train/jastadd/transformations/inject/JastaddTransformationInjectConnectedSegments.java b/trainbenchmark-tool-jastadd-specialized-base/src/main/java/de/tudresden/inf/st/train/jastadd/transformations/inject/JastaddTransformationInjectConnectedSegments.java
index a840cbf999b141cc601fbf2f11cf26ce7a29c092..bd762b99b9270fd0af84f763ed0593adc0e07bf5 100644
--- a/trainbenchmark-tool-jastadd-specialized-base/src/main/java/de/tudresden/inf/st/train/jastadd/transformations/inject/JastaddTransformationInjectConnectedSegments.java
+++ b/trainbenchmark-tool-jastadd-specialized-base/src/main/java/de/tudresden/inf/st/train/jastadd/transformations/inject/JastaddTransformationInjectConnectedSegments.java
@@ -31,12 +31,7 @@ public class JastaddTransformationInjectConnectedSegments<TJastaddDriver extends
       match.getSensor().addMonitoredElement(segment2);
 
       // remove the connection of segment1 to segment3
-      for (int i = 0; i < match.getSegment1().getConnectsToList().getNumChild(); i++) {
-        if (match.getSegment1().getConnectsTo(i) == match.getSegment3()) {
-          match.getSegment1().getConnectsToList().removeChild(i);
-          break;
-        }
-      }
+      match.getSegment1().removeConnectsTo(match.getSegment3());
 
       // connect segment1 to segment2
       match.getSegment1().addConnectsTo(segment2);