diff --git a/src/ptbench/scripts/dataset.py b/src/ptbench/scripts/dataset.py
index 5174abc8827df6e2f13bab7d5dbf48741482eeac..844311b5b19d2ec6637f5a153a15b5910387b12e 100644
--- a/src/ptbench/scripts/dataset.py
+++ b/src/ptbench/scripts/dataset.py
@@ -179,7 +179,7 @@ def check(dataset, limit):
                 logger.info(f"Checking protocol {protocol}")
 
                 errors += check_database_split_loading(
-                    database_split.subsets, raw_data_loader, limit=limit
+                    database_split._subsets, raw_data_loader, limit=limit
                 )
 
         if not errors: