diff --git a/tests/unit/bentoml_io/test_decorators.py b/tests/unit/bentoml_io/test_decorators.py
index 9d3597afbbe..fafd951e215 100644
--- a/tests/unit/bentoml_io/test_decorators.py
+++ b/tests/unit/bentoml_io/test_decorators.py
@@ -142,7 +142,7 @@ def pandas_func(
 
 
 def test_api_decorator_openapi_overrides():
-    from bentoml._internal.service import Service
+    from bentoml._internal.service.factory import Service
     from bentoml._internal.service.openapi import generate_spec
     from bentoml._internal.service.openapi.specification import OpenAPISpecification
 
@@ -188,7 +188,7 @@ def predict(self, data: str) -> str:
 
 
 def test_api_decorator_parameter_overrides():
-    from bentoml._internal.service import Service
+    from bentoml._internal.service.factory import Service
     from bentoml._internal.service.openapi import generate_spec
     from bentoml._internal.service.openapi.specification import OpenAPISpecification
 
@@ -244,7 +244,7 @@ def list_items(self, data: str) -> str:
 
 
 def test_api_decorator_response_overrides():
-    from bentoml._internal.service import Service
+    from bentoml._internal.service.factory import Service
     from bentoml._internal.service.openapi import generate_spec
     from bentoml._internal.service.openapi.specification import OpenAPISpecification
 
@@ -409,7 +409,7 @@ def test_api_decorator_invalid_overrides():
 
     import pytest
 
-    from bentoml._internal.service import Service
+    from bentoml._internal.service.factory import Service
     from bentoml._internal.service.openapi import generate_spec
 
     # Test invalid field name