diff --git a/config/subset_to_filenames.pkl b/config/subset_to_filenames.pkl
index 1317a29..3e6c2d8 100644
Binary files a/config/subset_to_filenames.pkl and b/config/subset_to_filenames.pkl differ
diff --git a/src/subset.py b/src/subset.py
index 288a152..ab3e198 100644
--- a/src/subset.py
+++ b/src/subset.py
@@ -9,7 +9,7 @@ def make_filename_to_subset_dict():
     for subset in range(10):
         files = glob.glob(in_folder+'subset{0}'.format(subset)+'/*.mhd')
         print "Subset", subset, "n files", len(files)
-        files = map(lambda x: x.split(x.replace('.mhd',''),'/')[-1], files)
+        files = map(lambda x: x.replace('.mhd','').split('/')[-1], files)
         subset_dict[subset] = files
 
     joblib.dump(subset_dict, '../config/subset_to_filenames.pkl')