diff --git a/TTS/tts/tf/utils/generic_utils.py b/TTS/tts/tf/utils/generic_utils.py
index e76893c2..91434a38 100644
--- a/TTS/tts/tf/utils/generic_utils.py
+++ b/TTS/tts/tf/utils/generic_utils.py
@@ -83,7 +83,7 @@ def setup_model(num_chars, num_speakers, c, enable_tflite=False):
         num_chars=num_chars,
         num_speakers=num_speakers,
         r=c.r,
-        postnet_output_dim=c.audio["num_mels"],
+        out_channels=c.audio["num_mels"],
         decoder_output_dim=c.audio["num_mels"],
         attn_type=c.attention_type,
         attn_win=c.windowing,
diff --git a/notebooks/PlotUmapLibriTTS.ipynb b/notebooks/PlotUmapLibriTTS.ipynb
index 97f9800d..0448f3df 100644
--- a/notebooks/PlotUmapLibriTTS.ipynb
+++ b/notebooks/PlotUmapLibriTTS.ipynb
@@ -33,7 +33,9 @@
     "from TTS.utils.audio import AudioProcessor\n",
     "from TTS.utils.io import load_config\n",
 =======
-    "from TTS.tts.utils.audio import AudioProcessor\n",
+    "from TTS.utils.audio import AudioProcessor
+
+\n",
     "from TTS.tts.utils.generic_utils import load_config\n",
 >>>>>>> dev
     "\n",
diff --git a/notebooks/TestAttention.ipynb b/notebooks/TestAttention.ipynb
index ed1c245b..5d8eed85 100644
--- a/notebooks/TestAttention.ipynb
+++ b/notebooks/TestAttention.ipynb
@@ -37,7 +37,9 @@
     "import librosa.display\n",
     "\n",
     "from TTS.tts.layers import *\n",
-    "from TTS.tts.utils.audio import AudioProcessor\n",
+    "from TTS.utils.audio import AudioProcessor
+
+\n",
     "from TTS.tts.utils.generic_utils import setup_model\n",
     "from TTS.tts.utils.io import load_config\n",
     "from TTS.tts.utils.text import text_to_sequence\n",