diff --git a/tensorflow_probability/examples/jupyter_notebooks/Gaussian_Process_Regression_In_TFP.ipynb b/tensorflow_probability/examples/jupyter_notebooks/Gaussian_Process_Regression_In_TFP.ipynb
index da832bd9ba..88cd3d07b4 100644
--- a/tensorflow_probability/examples/jupyter_notebooks/Gaussian_Process_Regression_In_TFP.ipynb
+++ b/tensorflow_probability/examples/jupyter_notebooks/Gaussian_Process_Regression_In_TFP.ipynb
@@ -291,6 +291,7 @@
         "import time\n",
         "\n",
         "import numpy as np\n",
+        "import matplotlib.pyplot as plt\n",
         "import tensorflow.compat.v2 as tf\n",
         "import tensorflow_probability as tfp\n",
         "tfb = tfp.bijectors\n",