diff --git a/tests/test_query_method.py b/tests/test_query_method.py
index 3741fcbf3..dd48e55d3 100644
--- a/tests/test_query_method.py
+++ b/tests/test_query_method.py
@@ -13,6 +13,9 @@ from fastapi import Depends, FastAPI
 from fastapi.testclient import TestClient
 from pydantic import BaseModel
 
+# Ensure compatibility across Python versions
+import sys
+
 
 def test_query_method_basic():
     app = FastAPI()