diff --git a/src/graphics/axleCounting/AxleCountingDrawAssistant.ts b/src/graphics/axleCounting/AxleCountingDrawAssistant.ts
index 9071f8e..c06ed97 100644
--- a/src/graphics/axleCounting/AxleCountingDrawAssistant.ts
+++ b/src/graphics/axleCounting/AxleCountingDrawAssistant.ts
@@ -170,10 +170,7 @@ export class AxleCountingDraw extends GraphicDrawAssistant<
 
     sections.forEach((section) => {
       const sectionRelations =
-        section.relationManage.getRelationsOfGraphicAndOtherType(
-          section,
-          Section.Type
-        );
+        section.relationManage.getRelationsOfGraphic(section);
       const ps = section.localToCanvasPoint(section.getStartPoint());
       const pe = section.localToCanvasPoint(section.getEndPoint());
       sectionRelations.forEach((relation) => {
@@ -209,7 +206,7 @@ export class AxleCountingDraw extends GraphicDrawAssistant<
         );
       const ps = section.localToCanvasPoint(section.getStartPoint());
       const pe = section.localToCanvasPoint(section.getEndPoint());
-      if (axleCountingRelations.length < 2) {
+      if (axleCountingRelations.length == 1) {
         axleCountingRelations.forEach((relation) => {
           const port = relation.getRelationParam(section).param;
           let addPort = SectionPort.A;