From bcd8d3a507905959ebc418574d5782531d6d8ccb Mon Sep 17 00:00:00 2001
From: Kaaninchen <124433727+kaaninchen@users.noreply.github.com>
Date: Fri, 17 Jul 2026 16:45:32 +0200
Subject: [PATCH] Fix leftover state mentions

---
 src/embeds.py   |  3 +--
 src/handlers.py | 13 +++++++++----
 src/utils.py    |  2 +-
 3 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/embeds.py b/src/embeds.py
index 3397e85..c3fab8a 100644
--- a/src/embeds.py
+++ b/src/embeds.py
@@ -20,9 +20,8 @@ def build_info_embed() -> discord.Embed | None:
 
     conn = handlers.current
     operator = operator_infos(conn["train"])
-    name = f"{conn['train']} nach {conn['destination']}"
     embed = discord.Embed(
-        title=name,
+        title=handlers.train_name,
         description=f"Abfahrt von {conn['station']} um {format_timestamp(conn['departure'])}",
         color=operator["color"]
     )
diff --git a/src/handlers.py b/src/handlers.py
index e4d8108..7c85a7a 100644
--- a/src/handlers.py
+++ b/src/handlers.py
@@ -5,7 +5,7 @@ from src.utils import random_connection
 current = None
 
 async def rename_vc(bot: discord.Bot) -> bool:
-    global current, name
+    global current, train_name
 
     guild = bot.get_guild(int(config["server"]))
     if guild is None:
@@ -18,9 +18,14 @@ async def rename_vc(bot: discord.Bot) -> bool:
         return False
     
     current = random_connection()
-    name = f"{current['train']} nach {current['destination']}"
 
-    print(f"Info: {name} from {current['station']} \n via: {current['via']}")
+    if current['train'].split()[0] == "IC" or current['train'].split()[0] == "ICE":
+        train = current['train']
+    else:
+        train = current['train'].split()[1]
+    train_name = f"{train} nach {current['destination']}"
 
-    await channel.edit(name=f"{config['formatting']}{name}",)
+    print(f"Info: {train_name} from {current['station']} \n via: {current['via']}")
+
+    await channel.edit(name=f"{config['formatting']}{train_name}",)
     return True
\ No newline at end of file
diff --git a/src/utils.py b/src/utils.py
index 170db15..5ec9615 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -39,7 +39,7 @@ def operator_infos(train):
         return OPERATORS["fallback"]
 
     train_type = train.split()[0]
-
+    print(train_type)
     return OPERATORS.get(train_type, OPERATORS["fallback"])
 
 def format_via_list(via: list[str]):