From 4787dabefc2851b26e0be1ebaa35f0c30408c7ba Mon Sep 17 00:00:00 2001
From: Spythere <spythere@protonmail.com>
Date: Thu, 25 Aug 2022 15:15:28 +0200
Subject: [PATCH] Poprawka masy ET41

---
 src/utils/vehicleUtils.ts | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/utils/vehicleUtils.ts b/src/utils/vehicleUtils.ts
index 8ad50fa..2897257 100644
--- a/src/utils/vehicleUtils.ts
+++ b/src/utils/vehicleUtils.ts
@@ -40,6 +40,10 @@ export function locoDataList(state: IStore) {
         ) {
           mass = 83;
         }
+
+        if (locoType.startsWith('ET')) {
+          mass = 167;
+        }
       }
 
       // Spalinowozy
@@ -157,10 +161,13 @@ export function chosenRealStock(state: IStore) {
     }, [] as string[])
     .join(';');
 
+  const realStockObj = Object.values(state.readyStockList).find(
+    (readyStock) => readyStock.stockString == currentStockString
+  );
 
-  const realStockObj = Object.values(state.readyStockList).find((readyStock) => readyStock.stockString == currentStockString);
-
-  state.chosenRealStockName = realStockObj ? `${realStockObj.type} ${realStockObj.number} ${realStockObj.name}` : undefined;
+  state.chosenRealStockName = realStockObj
+    ? `${realStockObj.type} ${realStockObj.number} ${realStockObj.name}`
+    : undefined;
 
   return realStockObj;
 }