diff --git a/src/allianceutils/middleware/query_count.py b/src/allianceutils/middleware/query_count.py
index 6ed88e1..a4536b2 100644
--- a/src/allianceutils/middleware/query_count.py
+++ b/src/allianceutils/middleware/query_count.py
@@ -69,8 +69,12 @@ def __call__(self, request: HttpRequest) -> HttpResponse:
 
     @classmethod
     def set_threshold(cls, request: HttpRequest, threshold: int):
-        cast(QueryCountHttpRequest, request).QUERY_COUNT_WARNING_THRESHOLD = threshold
+        # Use underlying Django request if available (e.g., DRF Request)
+        target_request = getattr(request, '_request', request)
+        cast(QueryCountHttpRequest, target_request).QUERY_COUNT_WARNING_THRESHOLD = threshold
 
     @classmethod
     def increase_threshold(cls, request: HttpRequest, increment: int):
-        cast(QueryCountHttpRequest, request).QUERY_COUNT_WARNING_THRESHOLD += increment
\ No newline at end of file
+        # Use underlying Django request if available (e.g., DRF Request)
+        target_request = getattr(request, '_request', request)
+        cast(QueryCountHttpRequest, target_request).QUERY_COUNT_WARNING_THRESHOLD += increment