From 3cd7201883a084b5ce8bb48033b26327b7c660de Mon Sep 17 00:00:00 2001
From: Spythere <spythere@protonmail.com>
Date: Sun, 6 Jul 2025 02:32:20 +0200
Subject: [PATCH] fix: 0 load for loco only

---
 src/mixins/useVehicleMixin.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/mixins/useVehicleMixin.ts b/src/mixins/useVehicleMixin.ts
index 4e65c99..48cc784 100644
--- a/src/mixins/useVehicleMixin.ts
+++ b/src/mixins/useVehicleMixin.ts
@@ -12,12 +12,13 @@ export function useVehicleMixin() {
     const stockArray = stockString.split(';');
     const headUnitsNames = stockArray.slice(0, 3).filter((v) => /-\d{3,}$/.test(v));
 
-    if (headUnitsNames.length == 1 && stockArray.length == 1) return trainMass;
-
     const headVehicleData = apiStore.vehiclesData.find((v) => v.name == headUnitsNames[0]);
-
+    
     if (!headVehicleData) return trainMass;
 
+    // 0t load for loco only
+    if (headVehicleData.type.startsWith("loco") && stockArray.length == 1) return 0;
+
     return Math.min(trainMass, trainMass - headVehicleData.group.weight);
   }