diff --git a/tests/test_pydantic_migrate.py b/tests/test_pydantic_migrate.py
index 56fcaa2ff..4779661ec 100644
--- a/tests/test_pydantic_migrate.py
+++ b/tests/test_pydantic_migrate.py
@@ -3,11 +3,9 @@
 
 import pytest
 from django.db import models
-
-from ninja import ModelSchema
 from pydantic import BaseModel, ValidationError
 
-from ninja import Schema
+from ninja import ModelSchema, Schema
 
 
 class OptModel(BaseModel):