diff --git a/TTS/vocoder/models/gan.py b/TTS/vocoder/models/gan.py
index 81ba87c4..1b49831e 100644
--- a/TTS/vocoder/models/gan.py
+++ b/TTS/vocoder/models/gan.py
@@ -310,6 +310,7 @@ class GAN(BaseVocoder):
         data_items: List,
         verbose: bool,
         num_gpus: int,
+        rank:int=0
     ):
         """Initiate and return the GAN dataloader.