diff --git a/Machine Learning/Data-Augmentation.ipynb b/Machine Learning/Data-Augmentation.ipynb
index d9ac38d..30a5575 100644
--- a/Machine Learning/Data-Augmentation.ipynb	
+++ b/Machine Learning/Data-Augmentation.ipynb	
@@ -68,6 +68,7 @@
     "import numpy as np\n",
     "import imgaug as ia\n",
     "import imgaug.augmenters as iaa\n",
+    "import matplotlib.pyplot as plt\n",
     "\n",
     "\n",
     "ia.seed(1)"