diff --git a/build.gradle.kts b/build.gradle.kts
index 2b86c40..0457b66 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -6,7 +6,7 @@ plugins {
 }
 
 group = "net.legacyfabric"
-version = "1.0.0"
+version = "1.0.1"
 
 repositories {
     mavenCentral()
diff --git a/src/main/java/net/legacyfabric/mcpmatcher/util/MappingUtils.java b/src/main/java/net/legacyfabric/mcpmatcher/util/MappingUtils.java
index 8b88601..340541d 100644
--- a/src/main/java/net/legacyfabric/mcpmatcher/util/MappingUtils.java
+++ b/src/main/java/net/legacyfabric/mcpmatcher/util/MappingUtils.java
@@ -43,12 +43,11 @@ private static void readJarFile(Path jarFile, SignatureVisitor visitor) throws I
     }
 
     public static MappingSet readMappingsFile(String arg) throws IOException {
-        //TODO: could use a map here and support more formats.
-        if (arg.endsWith(".csrg")) {
-            return MappingFormats.CSRG.read(Paths.get(arg));
-        } else {
-            return MappingFormats.SRG.read(Paths.get(arg));
-        }
+        return switch (arg.split("\\.")[1]) {
+            case "csrg" -> MappingFormats.CSRG.read(Paths.get(arg));
+            case "tsrg" -> MappingFormats.TSRG.read(Paths.get(arg));
+            default -> MappingFormats.SRG.read(Paths.get(arg));
+        };
     }
 
     private static class SignatureVisitor extends ClassVisitor {