diff --git a/litellm/router.py b/litellm/router.py
index fe8709294..b4603c6d0 100644
--- a/litellm/router.py
+++ b/litellm/router.py
@@ -63,7 +63,7 @@ class Router:
 
     def __init__(
         self,
-        model_list: Optional[List[DeploymentTypedDict]] = None,
+        model_list: Optional[List[Union[DeploymentTypedDict, Dict]]] = None,
         ## CACHING ##
         redis_url: Optional[str] = None,
         redis_host: Optional[str] = None,
diff --git a/litellm/tests/test_azure_perf.py b/litellm/tests/test_azure_perf.py
index 9654f1273..8afc59f92 100644
--- a/litellm/tests/test_azure_perf.py
+++ b/litellm/tests/test_azure_perf.py
@@ -26,7 +26,7 @@ model_list = [
     }
 ]
 
-router = litellm.Router(model_list=model_list)
+router = litellm.Router(model_list=model_list)  # type: ignore
 
 
 async def _openai_completion():