From c04f0b4dbbe8f2300f4f1bf42f62350888a92da5 Mon Sep 17 00:00:00 2001
From: Sourav Basu <sourav.basu@okta.com>
Date: Wed, 10 Jun 2026 19:56:59 +0530
Subject: [PATCH] Fix for incorrect return type and missing __all__ type

---
 src/auth0/management/__init__.py          | 1 +
 src/auth0/management/management_client.py | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/auth0/management/__init__.py b/src/auth0/management/__init__.py
index e96cb4b5..0cb284b6 100644
--- a/src/auth0/management/__init__.py
+++ b/src/auth0/management/__init__.py
@@ -6640,6 +6640,7 @@ def __dir__():
     "CspReportingEndpoints",
     "CspReportingInfrastructure",
     "CustomDomain",
+    "CustomDomainHeader",
     "CustomDomainCustomClientIpHeader",
     "CustomDomainCustomClientIpHeaderEnum",
     "CustomDomainProvisioningTypeEnum",
diff --git a/src/auth0/management/management_client.py b/src/auth0/management/management_client.py
index c6ea7599..421f9bda 100644
--- a/src/auth0/management/management_client.py
+++ b/src/auth0/management/management_client.py
@@ -5,6 +5,8 @@
 from json import dumps
 from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Union
 
+from .core.request_options import RequestOptions
+
 import httpx
 from .client import AsyncAuth0, Auth0
 from .token_provider import TokenProvider
@@ -36,7 +38,7 @@ def _enforce_custom_domain_whitelist(request: httpx.Request) -> None:
         del request.headers[CUSTOM_DOMAIN_HEADER]
 
 
-def CustomDomainHeader(domain: str) -> Dict[str, Any]:
+def CustomDomainHeader(domain: str) -> RequestOptions:
     """Create request options that set the Auth0-Custom-Domain header for a single request.
 
     When both a global custom_domain (set at client init) and a per-request