diff --git a/04_multilingual-ner.ipynb b/04_multilingual-ner.ipynb
index 203ec10..d543b86 100644
--- a/04_multilingual-ner.ipynb
+++ b/04_multilingual-ner.ipynb
@@ -2735,6 +2735,7 @@
    "outputs": [],
    "source": [
     "from sklearn.metrics import ConfusionMatrixDisplay, confusion_matrix\n",
+    "import matplotlib.pyplot as plt\n",
     "\n",
     "def plot_confusion_matrix(y_preds, y_true, labels):\n",
     "    cm = confusion_matrix(y_true, y_preds, normalize=\"true\")\n",