diff --git a/src/rail/core/data.py b/src/rail/core/data.py
index c64f1f3..677bb5c 100644
--- a/src/rail/core/data.py
+++ b/src/rail/core/data.py
@@ -282,7 +282,7 @@ def _size(self, path, **kwargs):
         if path in [None, 'none', 'None']:  # pragma: no cover
             return 0
         try:
-            return tables_io.io.getInputDataLengthHdf5(path, **kwargs)
+            return tables_io.io.getInputDataLength(path, **kwargs)
         except Exception:
             return 0
 
@@ -367,6 +367,9 @@ class PqHandle(TableHandle):
 
     suffix = "pq"
 
+    @classmethod
+    def _size(cls, path, **kwargs):
+        return tables_io.io.getInputDataLengthPq(path, **kwargs)
 
 class QPHandle(DataHandle):
     """DataHandle for qp ensembles"""