diff --git a/DeOldify_colab.ipynb b/DeOldify_colab.ipynb
index 2fb2eda..1300f28 100644
--- a/DeOldify_colab.ipynb
+++ b/DeOldify_colab.ipynb
@@ -118,7 +118,7 @@
     "pandocfilters>=1.4.2\n",
     "pexpect>=4.2.1\n",
     "pickleshare>=0.7.4\n",
-    "Pillow=4.2.1\n",
+    "Pillow>=5.3.0\n",
     "prompt-toolkit>=1.0.15\n",
     "Pygments>=2.2.0\n",
     "pyparsing>=2.2.0\n",
@@ -416,7 +416,7 @@
     "from pathlib import Path\n",
     "from itertools import repeat\n",
     "from google.colab import drive\n",
-    "from IPython.display import Image\n",
+    "from PIL import Image\n",
     "import tensorboardX\n",
     "torch.cuda.set_device(0)\n",
     "plt.style.use('dark_background')\n",