diff --git a/proto/FTCS.ipynb b/proto/FTCS.ipynb
index b3febbf..91d0acc 100644
--- a/proto/FTCS.ipynb
+++ b/proto/FTCS.ipynb
@@ -190,7 +190,7 @@
    "outputs": [],
    "source": [
     "def update(w = 1):\n",
-    "    fig = plt.imshow(figsize = (15,10))\n",
+    "    fig = plt.figure(figsize = (15,10))\n",
     "    y = records[w]\n",
     "    plt.imshow(y)\n",
     "    \n",