diff --git a/src/allencell_ml_segmenter/_tests/services/test_prediction_service.py b/src/allencell_ml_segmenter/_tests/services/test_prediction_service.py
index f7379f17..aa77e386 100644
--- a/src/allencell_ml_segmenter/_tests/services/test_prediction_service.py
+++ b/src/allencell_ml_segmenter/_tests/services/test_prediction_service.py
@@ -14,7 +14,6 @@
 from allencell_ml_segmenter.services.prediction_service import (
     PredictionService,
 )
-import cyto_dl
 
 
 @pytest.fixture
diff --git a/src/allencell_ml_segmenter/services/prediction_service.py b/src/allencell_ml_segmenter/services/prediction_service.py
index b7de39e5..07556456 100644
--- a/src/allencell_ml_segmenter/services/prediction_service.py
+++ b/src/allencell_ml_segmenter/services/prediction_service.py
@@ -8,7 +8,7 @@
 from allencell_ml_segmenter.prediction.model import PredictionModel
 from pathlib import Path
 from typing import List, Any
-from cyto_dl.api.model import CytoDLModel
+#from cyto_dl.api.model import CytoDLModel
 from napari.utils.notifications import show_warning
 
 
diff --git a/src/allencell_ml_segmenter/services/training_service.py b/src/allencell_ml_segmenter/services/training_service.py
index 78def9dd..b4f92ade 100644
--- a/src/allencell_ml_segmenter/services/training_service.py
+++ b/src/allencell_ml_segmenter/services/training_service.py
@@ -3,7 +3,7 @@
 from allencell_ml_segmenter.core.subscriber import Subscriber
 from allencell_ml_segmenter.core.event import Event
 
-from cyto_dl.api.model import CytoDLModel
+#from cyto_dl.api.model import CytoDLModel
 
 # from lightning.pytorch.callbacks import Callback