diff --git a/configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml b/configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml
index 6a63ab1..f3e4519 100644
--- a/configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml
+++ b/configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml
@@ -28,6 +28,7 @@ trainer:
   max_epochs: 310
   precision: 16
   devices: 8
+  use_distributed_sampler: ${eval:"${dataset.num_aug_repeats} == 0"}  # only true if using RepeatAug
 
 train:
   seed: 1112