diff --git a/tests/endpoints/a.py b/tests/endpoints/a.py
index be5663f59..19a1b18cc 100644
--- a/tests/endpoints/a.py
+++ b/tests/endpoints/a.py
@@ -1,4 +1,4 @@
-from fastapi.routing import APIRouter
+from fastapi import APIRouter
 
 router = APIRouter()
 
diff --git a/tests/endpoints/b.py b/tests/endpoints/b.py
index 7747fb2e4..9105a7144 100644
--- a/tests/endpoints/b.py
+++ b/tests/endpoints/b.py
@@ -1,4 +1,4 @@
-from fastapi.routing import APIRouter
+from fastapi import APIRouter
 
 router = APIRouter()
 
diff --git a/tests/main.py b/tests/main.py
index 7c4e639fd..6795baae4 100644
--- a/tests/main.py
+++ b/tests/main.py
@@ -1,22 +1,16 @@
-from fastapi.applications import FastAPI
-from fastapi.params import (
+from fastapi.applications import (
     Body,
     Cookie,
     Depends,
+    FastAPI,
     File,
     Form,
     Header,
-    Param,
     Path,
     Query,
     Security,
 )
-from fastapi.security.http import HTTPBasic
-from fastapi.security.oauth2 import (
-    OAuth2,
-    OAuth2PasswordRequestData,
-    OAuth2PasswordRequestForm,
-)
+from fastapi.security import HTTPBasic, OAuth2, OAuth2PasswordRequestForm
 from pydantic import BaseModel
 from starlette.responses import HTMLResponse, JSONResponse, PlainTextResponse
 from starlette.status import HTTP_202_ACCEPTED