diff --git a/src/main.py b/src/main.py
index 4f381bb..ef404b9 100644
--- a/src/main.py
+++ b/src/main.py
@@ -4,7 +4,7 @@ import argparse
 import json
 from os.path import exists
 
-from fastapi import FastAPI, Request, HTTPException
+from fastapi import FastAPI, Request, HTTPException, Query
 from fastapi.responses import StreamingResponse
 from fastapi import status
 from hypercorn.config import Config
@@ -44,7 +44,7 @@ class MinSizePerRequestError(Exception):
 
 
 @api.get('/')
-def test_data(api_key: str, size: str, request: Request) -> StreamingResponse:
+def test_data(request: Request, size: str, api_key: str = Query(..., alias="api-key")) -> StreamingResponse:
     try:
         if api_key not in AUTHORIZED_KEYS:
             raise HTTPException(