diff --git a/tests/base/test_pipelines.py b/tests/base/test_pipelines.py
index 3c643c0a8faac926b9303c0c1edf2ad34494ca27..a7ccb9975773e416a03d220c849c1f1f78dadb33 100644
--- a/tests/base/test_pipelines.py
+++ b/tests/base/test_pipelines.py
@@ -2,7 +2,7 @@ from pathlib import Path
 import unittest
 
 from model_server.base.accessors import generate_file_accessor, write_accessor_data_to_file
-from model_server.base.pipelines import segment, segment_zproj
+from model_server.base.pipelines import router, segment, segment_zproj
 
 import model_server.conf.testing as conf
 from tests.base.test_model import DummySemanticSegmentationModel