diff --git a/khaosz/data/sampler.py b/khaosz/data/sampler.py
index 4a7ddbc..114afd3 100644
--- a/khaosz/data/sampler.py
+++ b/khaosz/data/sampler.py
@@ -75,6 +75,4 @@ class ResumableDistributedSampler(Sampler[int]):
         self._indices = None
     
     def __len__(self):
-        if self._indices is None:
-            self._get_indices()
-        return len(self._indices)
\ No newline at end of file
+        return self.num_samples_per_replica
\ No newline at end of file