From c3bfe08364a8f82b508c98a0cbb90b15c317bfe9 Mon Sep 17 00:00:00 2001
From: Scirlat Danut <scirlatdanut@scirlats-mini.lan>
Date: Sun, 3 Mar 2024 18:59:24 +0200
Subject: [PATCH] Small improvements

---
 tests/types.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/types.py b/tests/types.py
index 383d52b9f..4ab704159 100644
--- a/tests/types.py
+++ b/tests/types.py
@@ -2,7 +2,7 @@
 
 from typing import Protocol
 
-from httpx import Cookies, Headers
+import httpx
 
 from starlette.testclient import TestClient
 from starlette.types import ASGIApp
@@ -17,8 +17,8 @@ def __call__(
         base_url: str = "http://testserver",
         raise_server_exceptions: bool = True,
         root_path: str = "",
-        cookies: Cookies | dict[str, str] | None = None,
-        headers: Headers | None = None,
+        cookies: httpx._types.CookieTypes | None = None,
+        headers: dict[str, str] | None = None,
         follow_redirects: bool = True,
     ) -> TestClient:
         ...