diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
index 23be6177..1de26913 100644
--- a/TTS/tts/models/vits.py
+++ b/TTS/tts/models/vits.py
@@ -275,10 +275,7 @@ class Vits(BaseTTS):
     # pylint: disable=dangerous-default-value
 
     def __init__(
-        self,
-        config: Coqpit,
-        speaker_manager: SpeakerManager = None,
-        language_manager: LanguageManager = None,
+        self, config: Coqpit, ap: "AudioProcessor", tokenizer: "TTSTokenizer", speaker_manager: SpeakerManager = None, language_manager: LanguageManager = None
     ):
 
         super().__init__(config, ap, tokenizer, speaker_manager)