diff --git a/NiChart_DLMUSE/utils.py b/NiChart_DLMUSE/utils.py
index f23825c..477dfe7 100644
--- a/NiChart_DLMUSE/utils.py
+++ b/NiChart_DLMUSE/utils.py
@@ -176,6 +176,9 @@ def split_data(in_dir: str, N: int) -> list:
             os.system(f"cp {file} {in_dir}/split_{current_folder}")
             current_file += 1
 
+    if current_file > 0:
+        subfolders.append(f"{in_dir}/split_{current_folder}")
+
     return subfolders