diff --git a/test/drs_import.py b/test/drs_import.py
index b62bf7cd2d190f45aa1c7089ec10815d23a1b9f5..694d9b3b489e931f4958d9e2b8edf3413711c1a7 100644
--- a/test/drs_import.py
+++ b/test/drs_import.py
@@ -1,5 +1,6 @@
 import spot
 import argparse
+import os
 from glob import glob
 
 # Arguments
@@ -9,7 +10,7 @@ params = parser.parse_args()
 
 # List files
 print("List files...")
-files = glob("{}/*/".format(params.root_dir), recursive=True)
+files = [y for x in os.walk(params.root_dir) for y in glob(os.path.join(x[0], "DIM*.XML"))]
 print(len(files))
 
 for i in range(0,100):