diff --git a/chooseK.py b/chooseK.py
index d448e2b..07e0a04 100644
--- a/chooseK.py
+++ b/chooseK.py
@@ -102,7 +102,7 @@ def usage():
     filetag = parseopts(opts)
 
     files = glob.glob('%s.*.log'%filetag)
-    Ks = np.array([int(file.split('.')[1]) for file in files])
+    Ks = np.array([int(file.split('.')[-2]) for file in files])
     marginal_likelihoods = parse_logs(files)
 
     files = glob.glob('%s.*.meanQ'%filetag)