diff --git a/examples/colab_record_example.ipynb b/examples/colab_record_example.ipynb
index 7658770..1f425d4 100644
--- a/examples/colab_record_example.ipynb
+++ b/examples/colab_record_example.ipynb
@@ -141,7 +141,7 @@
     "    fig = plt.figure(figsize=(16, 9))\n",
     "    ax = plt.axes(xlim=(0, audio_duration), ylim=(0, 1.02))\n",
     "    line, = ax.plot([], [], lw=2)\n",
-    "    x = [i / 16000 * 1536 for i in range(len(probs))]\n",
+    "    x = [i / 16000 * 512 for i in range(len(probs))]\n",
     "    plt.xlabel('Time, seconds', fontsize=16)\n",
     "    plt.ylabel('Speech Probability', fontsize=16)\n",
     "\n",
@@ -152,7 +152,7 @@
     "        return line,\n",
     "\n",
     "    def animate(i):\n",
-    "        x = i * interval / 1000 - 0.1\n",
+    "        x = i * interval / 1000 - 0.04\n",
     "        y = np.linspace(0, 1.02, 2)\n",
     "        \n",
     "        line.set_data(x, y)\n",
@@ -177,7 +177,7 @@
     "\n",
     "  print('Calculating probabilities...')\n",
     "  speech_probs = []\n",
-    "  window_size_samples = 1536\n",
+    "  window_size_samples = 512\n",
     "  for i in range(0, len(tensor), window_size_samples):\n",
     "      if len(tensor[i: i+ window_size_samples]) < window_size_samples:\n",
     "        break\n",