diff --git a/tensorflow-mnist-tutorial/keras_01_mnist.ipynb b/tensorflow-mnist-tutorial/keras_01_mnist.ipynb
index 2ccd191..b28d202 100644
--- a/tensorflow-mnist-tutorial/keras_01_mnist.ipynb
+++ b/tensorflow-mnist-tutorial/keras_01_mnist.ipynb
@@ -156,7 +156,7 @@
     "  digits = np.reshape(digits, [28, 28*n])\n",
     "  plt.yticks([])\n",
     "  plt.xticks([28*x+14 for x in range(n)], predictions)\n",
-    "  plt.grid(b=None)\n",
+    "  plt.grid(visible=None)\n",
     "  for i,t in enumerate(plt.gca().xaxis.get_ticklabels()):\n",
     "    if predictions[i] != labels[i]: t.set_color('red') # bad predictions in red\n",
     "  plt.imshow(digits)\n",