diff --git a/tests/test_jsonschema/test_jsonschema_generation.py b/tests/test_jsonschema/test_jsonschema_generation.py
index 2d55841..9ffd4a0 100644
--- a/tests/test_jsonschema/test_jsonschema_generation.py
+++ b/tests/test_jsonschema/test_jsonschema_generation.py
@@ -65,7 +65,6 @@
 from mashumaro.jsonschema.builder import JSONSchemaBuilder, build_json_schema
 from mashumaro.jsonschema.dialects import DRAFT_2020_12, OPEN_API_3_1
 from mashumaro.jsonschema.models import (
-    BasePlugin,
     Context,
     JSONArraySchema,
     JSONObjectSchema,
@@ -75,6 +74,7 @@
     JSONSchemaInstanceType,
     JSONSchemaStringFormat,
 )
+from mashumaro.jsonschema.plugins import BasePlugin
 from mashumaro.jsonschema.schema import (
     UTC_OFFSET_PATTERN,
     EmptyJSONSchema,