diff --git a/src/ptbench/data/dataset.py b/src/ptbench/data/dataset.py
index 07f18acf8b1ed3c68be535c621d371f026c29ff7..5966f392a7c66f60aeb089f92f8d88b03738a384 100644
--- a/src/ptbench/data/dataset.py
+++ b/src/ptbench/data/dataset.py
@@ -10,9 +10,9 @@ import pathlib
 import random
 
 import torch
-import tqdm
 
 from torchvision.transforms import RandomRotation
+from tqdm import tqdm
 
 RANDOM_ROTATION = [RandomRotation(15)]
 """Shared data augmentation based on random rotation only."""
@@ -179,7 +179,7 @@ class JSONDataset:
                     dict(zip(self.fieldnames, k)),
                     self.post_transforms,
                 )
-                for n, k in tqdm.tqdm(enumerate(samples))
+                for n, k in enumerate(tqdm(samples))
             ]
 
         return retval