diff --git a/trapdata/api/tests/test_api.py b/trapdata/api/tests/test_api.py
index 622310e..6069f98 100644
--- a/trapdata/api/tests/test_api.py
+++ b/trapdata/api/tests/test_api.py
@@ -7,12 +7,11 @@
 from trapdata.api.api import (
     PIPELINE_CHOICES,
     PipelineChoice,
-    PipelineConfig,
     PipelineRequest,
     PipelineResponse,
-    SourceImageRequest,
     app,
 )
+from trapdata.api.schemas import PipelineConfig, SourceImageRequest
 from trapdata.api.tests.image_server import StaticFileTestServer
 from trapdata.ml.models.classification import SpeciesClassifier
 from trapdata.tests import TEST_IMAGES_BASE_PATH