diff --git a/src/backend/base/langflow/components/tools/astradb.py b/src/backend/base/langflow/components/tools/astradb.py
index 2153b8082..432add9de 100644
--- a/src/backend/base/langflow/components/tools/astradb.py
+++ b/src/backend/base/langflow/components/tools/astradb.py
@@ -1,3 +1,4 @@
+import os
 from typing import Any
 
 from astrapy import Collection, DataAPIClient, Database
@@ -10,7 +11,7 @@ from langflow.schema import Data
 
 
 class AstraDBToolComponent(LCToolComponent):
-    display_name: str = "Astra DB"
+    display_name: str = "Astra DB Tool"
     description: str = "Create a tool to get transactional data from DataStax Astra DB Collection"
     documentation: str = "https://docs.langflow.org/Components/components-tools#astra-db-tool"
     icon: str = "AstraDB"
@@ -48,9 +49,9 @@ class AstraDBToolComponent(LCToolComponent):
             value="ASTRA_DB_APPLICATION_TOKEN",
             required=True,
         ),
-        StrInput(
+        SecretStrInput(
             name="api_endpoint",
-            display_name="API Endpoint",
+            display_name="Database" if os.getenv("ASTRA_ENHANCED", "false").lower() == "true" else "API Endpoint",
             info="API endpoint URL for the Astra DB service.",
             value="ASTRA_DB_API_ENDPOINT",
             required=True,