diff --git a/tests/test_model.py b/tests/test_model.py
index 57e07c8e16ad079d19f106517d9d06a9ef6daa51..91043f24e7ec354759dafa2c61b833b1dbd78688 100644
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -1,5 +1,5 @@
 import unittest
-from conf.testing import czifile
+from model_server.conf.testing import czifile
 from model_server.base.accessors import CziImageFileAccessor
 from model_server.base.models import DummySemanticSegmentationModel, DummyInstanceSegmentationModel, CouldNotLoadModelError
 
diff --git a/tests/test_workflow.py b/tests/test_workflow.py
index 19bba960caf3b35bbf4bad74123b1ee36c9f651d..6e9603ea7418ba367b482a00b8dd8c9aafa8820d 100644
--- a/tests/test_workflow.py
+++ b/tests/test_workflow.py
@@ -1,6 +1,6 @@
 import unittest
 
-from conf.testing import czifile, output_path
+from model_server.conf.testing import czifile, output_path
 from model_server.base.models import DummySemanticSegmentationModel
 from model_server.base.workflows import classify_pixels