diff --git a/testanalysis.py b/testanalysis.py
index 58f62c8..e91d121 100644
--- a/testanalysis.py
+++ b/testanalysis.py
@@ -26,7 +26,9 @@ def linesplit(line):
     dm = int(dm)
     pw, fname = line.split(" s ")
     pw = float(pw)
-    fname = fname.strip().removesuffix("_injected")
+    fname = fname.strip().removesuffix(".fil").removesuffix("_injected")
+    if fname.startswith("/"):
+        fname = fname.split("/")[-1]
     return (dm, pw, fname)
 
 def updateDic(lines, dic):