diff --git a/coverage-tool/coverage_tool.py b/coverage-tool/coverage_tool.py
index ea73dba..361f558 100644
--- a/coverage-tool/coverage_tool.py
+++ b/coverage-tool/coverage_tool.py
@@ -26,7 +26,7 @@
 from sys import stderr
 from urllib.request import urlopen
 
-DISCOVERY_DOC_URL = "https://aiplatform.googleapis.com/$discovery/rest?version=v1beta1"
+DISCOVERY_DOC_URL = "https://firebasevertexai.googleapis.com/$discovery/rest?version=v1beta1"
 SCHEMA_PREFIX = "GoogleCloudAiplatformV1beta1"
 RESPONSE_TYPES = ["GenerateContentResponse", "CountTokensResponse"]
 SCRIPT_DIR = os.path.dirname(__file__)