diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py
index 3ac753d2..36b69812 100644
--- a/examples/new_project_templates/lightning_module_template.py
+++ b/examples/new_project_templates/lightning_module_template.py
@@ -164,7 +164,7 @@ class LightningTemplateModel(LightningModule):
 
         # try:
         #     if self.on_gpu:
-        train_sampler = DistributedSampler(dataset, num_replicas=self.trainer.world_size, rank=self.trainer.proc_rank)
+        train_sampler = DistributedSampler(dataset, rank=self.trainer.proc_rank)
         batch_size = batch_size // self.trainer.world_size  # scale batch size
 
         # except Exception as e: