diff --git a/tts_pytorch.ipynb b/tts_pytorch.ipynb
index e2702e6..57f7447 100644
--- a/tts_pytorch.ipynb
+++ b/tts_pytorch.ipynb
@@ -81,7 +81,7 @@
     "pcms = []\n",
     "def _on_frame(frame):\n",
     "    print(\"Step\", len(pcms), end=\"\\r\")\n",
-    "    if (frame != -1).all():\n",
+    "    if (frame != -1).all().item():\n",
     "        pcm = tts_model.mimi.decode(frame[:, 1:, :]).cpu().numpy()\n",
     "        pcms.append(np.clip(pcm[0, 0], -1, 1))\n",
     "\n",