diff --git a/ppcls/data/dataloader/pk_sampler.py b/ppcls/data/dataloader/pk_sampler.py
index b3632bc72582243a8b041a4fd7b6b3fa1f0e7b88..bf563a6c14cf47f24efbae69fa1cdb5ab5c8640e 100644
--- a/ppcls/data/dataloader/pk_sampler.py
+++ b/ppcls/data/dataloader/pk_sampler.py
@@ -80,10 +80,7 @@ class PKSampler(DistributedBatchSampler):
 
     def __iter__(self):
         label_per_batch = self.batch_size // self.sample_per_label
-        if self.shuffle:
-            # It's not accurate literally, but it helps in some dataset.
-            np.random.RandomState(self.epoch).shuffle(self.label_list)
-        for i in range(len(self)):
+        for _ in range(len(self)):
             batch_index = []
             batch_label_list = np.random.choice(
                 self.label_list,