diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
index 751187ea..7dac1bb9 100644
--- a/TTS/tts/models/vits.py
+++ b/TTS/tts/models/vits.py
@@ -947,8 +947,8 @@ class Vits(BaseTTS):
             Tuple[Dict, np.ndarray]: training plots and output waveform.
         """
         figures, audios = self._log(self.ap, batch, outputs, "train")
-        logger.eval_figures(steps, figures)
-        logger.eval_audios(steps, audios, self.ap.sample_rate)
+        logger.train_figures(steps, figures)
+        logger.train_figures(steps, audios, self.ap.sample_rate)
 
     @torch.no_grad()
     def eval_step(self, batch: dict, criterion: nn.Module, optimizer_idx: int):