diff --git a/src/TrainsTable.js b/src/TrainsTable.js
index 1e103f1..463e02e 100644
--- a/src/TrainsTable.js
+++ b/src/TrainsTable.js
@@ -202,9 +202,13 @@ function getTrainType(train) {
         default:
           return trainType
       }
-    case "FR-IDF-IDFM":
+    case "FR-IDFM":
+      const route_split = train.routeShortName.split(" ")
+      if (route_split[0] === "Bus")
+        return route_split[1]
+      return route_split[0]
     case "FR-GES-CTS":
-      return "A"
+      return train.routeShortName.split(" ")[1]
     case "FR-EUROSTAR":
       return "Eurostar"
     case "IT-FRA-TI":
@@ -216,7 +220,6 @@ function getTrainType(train) {
         return "NJ"
       return "ÖBB"
     case "CH-ALL":
-      return "A"
     default:
       return train.routeShortName?.split(" ")[0]
   }