diff --git a/conditional_neural_process.ipynb b/conditional_neural_process.ipynb
index b300c80..4d4f4ae 100644
--- a/conditional_neural_process.ipynb
+++ b/conditional_neural_process.ipynb
@@ -688,7 +688,7 @@
         "  plt.ylim([-2, 2])\n",
         "  plt.grid('off')\n",
         "  ax = plt.gca()\n",
-        "  ax.set_axis_bgcolor('white')\n",
+        "  ax.set_facecolor('white')\n",
         "  plt.show()"
       ]
     },