diff --git a/botiverse/models/FastSpeech1/FastSpeech.ipynb b/botiverse/models/FastSpeech1/FastSpeech.ipynb
index c50f405..91efac1 100644
--- a/botiverse/models/FastSpeech1/FastSpeech.ipynb
+++ b/botiverse/models/FastSpeech1/FastSpeech.ipynb
@@ -1014,9 +1014,9 @@
     "            \n",
     "        if play:\n",
     "            write('sample.wav', 22050, audio.astype('int16'))\n",
-    "            playsound(\"sample.wav\")                         # comment me if on Google Colab\n",
-    "            #from IPython.display import Audio, display     # uncomment me if on Google Colab\n",
-    "            #display(Audio('sample.wav', autoplay=True))    # uncomment me if on Google Colab\n",
+    "            #playsound(\"sample.wav\")                       # comment me if on Google Colab\n",
+    "            from IPython.display import Audio, display     # comment me if running offline\n",
+    "            display(Audio('sample.wav', autoplay=True))    # comment me if running offline\n",
     "            os.remove(\"sample.wav\")\n",
     "        return audio"
    ]