diff --git a/TTS/trainer.py b/TTS/trainer.py
index 49f6a58c..d05028e2 100644
--- a/TTS/trainer.py
+++ b/TTS/trainer.py
@@ -812,7 +812,7 @@ class Trainer:
                 samples = self.eval_loader.dataset.load_test_samples(1)
                 figures, audios = self.model.test_run(self.ap, samples, None)
             else:
-                figures, audios = self.model.test_run()
+                figures, audios = self.model.test_run(self.ap)
             self.dashboard_logger.test_audios(self.total_steps_done, audios, self.config.audio["sample_rate"])
             self.dashboard_logger.test_figures(self.total_steps_done, figures)