diff --git a/tests/test_audio.py b/tests/test_audio.py
index fc5deb48..7f884d37 100644
--- a/tests/test_audio.py
+++ b/tests/test_audio.py
@@ -33,7 +33,7 @@ class TestAudio(unittest.TestCase):
             self.ap.clip_norm = clip_norm
             wav = self.ap.load_wav(WAV_FILE)
             mel = self.ap.melspectrogram(wav)
-            wav_ = self.ap.inv_mel_spectrogram(mel)
+            wav_ = self.ap.inv_melspectrogram(mel)
             file_name = "/audio_test-melspec_max_norm_{}-signal_norm_{}-symmetric_{}-clip_norm_{}.wav"\
                 .format(max_norm, signal_norm, symmetric_norm, clip_norm)
             print(" | > Creating wav file at : ", file_name)
diff --git a/train.py b/train.py
index 1397b310..f0e3b68a 100644
--- a/train.py
+++ b/train.py
@@ -13,7 +13,7 @@ from torch.utils.data import DataLoader
 from TTS.datasets.TTSDataset import MyDataset
 from distribute import (DistributedSampler, apply_gradient_allreduce,
                         init_distributed, reduce_tensor)
-from TTS.layers.losses import L1LossMasked, MSELossMasked, BCELossMasked
+from TTS.layers.losses import TacotronLoss
 from TTS.utils.audio import AudioProcessor
 from TTS.utils.generic_utils import (
     NoamLR, check_update, count_parameters, create_experiment_folder,