diff --git a/src/allencell_ml_segmenter/core/prediction_result_input_widget.py b/src/allencell_ml_segmenter/core/prediction_result_input_widget.py
index 9dd80bcd..e846f15e 100644
--- a/src/allencell_ml_segmenter/core/prediction_result_input_widget.py
+++ b/src/allencell_ml_segmenter/core/prediction_result_input_widget.py
@@ -30,10 +30,9 @@ def _update_layer_list(self) -> None:
         for idx, prediction_output_layer in enumerate(self._prediction_layers):
             self._image_list.add_item(
                 prediction_output_layer.name,
-                set_checked=idx in previous_selections
+                set_checked=idx in previous_selections,
             )
 
-
     def _process_checked_signal(self, row: int, state: Qt.CheckState) -> None:
         if self._model.get_input_mode() == InputMode.FROM_NAPARI_LAYERS:
             self._model.set_selected_idx(self._image_list.get_checked_rows())