diff --git a/cellfinder/napari/detect/detect.py b/cellfinder/napari/detect/detect.py
index 66bbbb22..07adaf0b 100644
--- a/cellfinder/napari/detect/detect.py
+++ b/cellfinder/napari/detect/detect.py
@@ -256,7 +256,7 @@ def widget(
         use_pre_trained_weights: bool,
         weights_only: bool,
         trained_model: Optional[Path],
-        model_weights,
+        model_weights: Optional[Path],
         batch_size: int,
         misc_options,
         start_plane: int,
diff --git a/cellfinder/napari/detect/detect_containers.py b/cellfinder/napari/detect/detect_containers.py
index 28f45961..4dfda0c2 100644
--- a/cellfinder/napari/detect/detect_containers.py
+++ b/cellfinder/napari/detect/detect_containers.py
@@ -133,6 +133,7 @@ def widget_representation(cls) -> dict:
             ),
             weights_only=dict(value=cls.defaults()["weights_only"]),
             trained_model=dict(value=cls.defaults()["trained_model"]),
+            model_weights=dict(value=cls.defaults()["model_weights"]),
             skip_classification=dict(
                 value=cls.defaults()["skip_classification"]
             ),