diff --git a/16_nlp_with_rnns_and_attention.ipynb b/16_nlp_with_rnns_and_attention.ipynb
index 7e8afa7..ad5963a 100644
--- a/16_nlp_with_rnns_and_attention.ipynb
+++ b/16_nlp_with_rnns_and_attention.ipynb
@@ -1140,7 +1140,7 @@
     "ax2.axis([0, max_steps - 1, 0, crop_i])\n",
     "ax2.set_xlabel(\"$p$\", fontsize=16)\n",
     "ax2.set_ylabel(\"$i$\", rotation=0, fontsize=16)\n",
-    "plt.savefig(\"positional_embedding_plot\")\n",
+    "save_fig(\"positional_embedding_plot\")\n",
     "plt.show()"
    ]
   },