From 7aaf620d6a5146bc5c07ff040f8ac71ceeafbc2e Mon Sep 17 00:00:00 2001
From: Spythere <spythere@protonmail.com>
Date: Sat, 3 May 2025 13:47:51 +0200
Subject: [PATCH] fix: head units regex

---
 src/utils/trainUtils.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/utils/trainUtils.ts b/src/utils/trainUtils.ts
index 217b38b..602a71b 100644
--- a/src/utils/trainUtils.ts
+++ b/src/utils/trainUtils.ts
@@ -32,7 +32,7 @@ export function getHeadUnits(stockString: string) {
   const stockList = stockString.split(';').slice(0, 3);
 
   return stockList.reduce((acc, unitType, i) => {
-    if (i != 0 && !/-\d+$/.test(unitType)) return acc;
+    if (i != 0 && !/-\d{3,}$/.test(unitType)) return acc;
 
     const unitName = unitType.slice(0, unitType.indexOf('-'));