diff --git a/16_reinforcement_learning.ipynb b/16_reinforcement_learning.ipynb
index e666ce2..ef93744 100644
--- a/16_reinforcement_learning.ipynb
+++ b/16_reinforcement_learning.ipynb
@@ -11348,7 +11348,7 @@
    "source": [
     "frames = run_bipedal_walker()\n",
     "video = plot_animation(frames)\n",
-    "plt.show()"
+    "HTML(video.to_html5_video())"
    ]
   },
   {
@@ -11412,7 +11412,7 @@
    "source": [
     "frames = run_bipedal_walker(\"./my_bipedal_walker_pg.ckpt\")\n",
     "video = plot_animation(frames)\n",
-    "plt.show()"
+    "HTML(video.to_html5_video())"
    ]
   },
   {
@@ -11447,7 +11447,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.5"
+   "version": "3.6.5"
   },
   "nav_menu": {},
   "toc": {