diff --git a/flooding/sentinel2_water_extraction/sentinel2_water_extraction.ipynb b/flooding/sentinel2_water_extraction/sentinel2_water_extraction.ipynb
index b8591eb..d4be9f7 100644
--- a/flooding/sentinel2_water_extraction/sentinel2_water_extraction.ipynb
+++ b/flooding/sentinel2_water_extraction/sentinel2_water_extraction.ipynb
@@ -96,7 +96,7 @@
     "# suppress warnings - raised by external rasterio library\n",
     "import warnings\n",
     "from datetime import date\n",
-    "from typing import List, Optional, Tuple, TypedDict\n",
+    "from typing import Annotated, List, Optional, Tuple, TypedDict\n",
     "\n",
     "import geopandas as gpd\n",
     "import matplotlib.pyplot as plt\n",
@@ -112,11 +112,6 @@
     "from shapely.geometry import shape\n",
     "from typing_extensions import NotRequired\n",
     "\n",
-    "if sys.version_info >= (3, 9):\n",
-    "    from typing import Annotated\n",
-    "else:\n",
-    "    from typing_extensions import Annotated\n",
-    "\n",
     "\n",
     "class Image(TypedDict):\n",
     "    \"\"\"Image info being passed around this code.\"\"\"\n",
diff --git a/pyproject.toml b/pyproject.toml
index 1b444f9..3563451 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,8 +9,6 @@ strict = true
 module = [
     "affine", # https://github.com/rasterio/affine/issues/74
     "geopandas", # https://github.com/geopandas/geopandas/issues/1974
-    "matplotlib", # https://github.com/matplotlib/matplotlib/issues/20504
-    "matplotlib.pyplot", # https://github.com/matplotlib/matplotlib/issues/20504
     "rasterio", # https://github.com/rasterio/rasterio/issues/2322
     "rasterio.plot", # https://github.com/rasterio/rasterio/issues/2322
     "satsearch", # https://github.com/sat-utils/sat-search/issues/131