diff --git a/dacapo/experiments/datasplits/datasplit_generator.py b/dacapo/experiments/datasplits/datasplit_generator.py
index 5858b2a6e..2f860bdbf 100644
--- a/dacapo/experiments/datasplits/datasplit_generator.py
+++ b/dacapo/experiments/datasplits/datasplit_generator.py
@@ -8,6 +8,7 @@
 from zarr.n5 import N5FSStore
 import numpy as np
 from dacapo.experiments.datasplits.datasets.arrays import (
+    ArrayConfig,
     ZarrArrayConfig,
     ZarrArray,
     ResampledArrayConfig,
@@ -916,8 +917,8 @@ def __generate_semantic_seg_dataset_crop(self, dataset: DatasetSpec):
             current_targets = self.targets
             targets_str = "_".join(self.targets)
 
-        target_images = {}
-        target_masks = {}
+        target_images = dict[str, ArrayConfig]()
+        target_masks = dict[str, ArrayConfig]()
 
         missing_classes = [c for c in current_targets if c not in classes]
         found_classes = [c for c in current_targets if c in classes]
diff --git a/pyproject.toml b/pyproject.toml
index 866c9fffe..1c5b726db 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -58,6 +58,7 @@ dependencies = [
         "scipy",
         "upath",
         "boto3",
+        "matplotlib",
         ]
 
 # extras
@@ -78,7 +79,6 @@ docs = [
     "sphinx-click",
     "sphinx-rtd-theme",
     "myst-parser",
-    "matplotlib",
     "pooch",
 ]
 examples = [