diff --git a/physgrad-code.ipynb b/physgrad-code.ipynb
index 7b26042..b92f807 100644
--- a/physgrad-code.ipynb
+++ b/physgrad-code.ipynb
@@ -564,7 +564,7 @@
    ],
    "source": [
     "plt = vis.plot(x_gd.values.batch[0], x_sip.values.batch[0], data.values.batch[0], size=(15,4) )\n",
-    "plt.get_axes()[0].set_title(\"Adam\"); plt.get_axes()[1].set_title(\"SIP\"); plt.get_axes()[2].set_title(\"Reference\");\n"
+    "plt.axes[0].title=\"Adam\"; plt.axes[1].title=\"SIP\"; plt.axes[2].title=\"Reference\";\n"
    ]
   },
   {
@@ -605,4 +605,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 0
-}
\ No newline at end of file
+}