diff --git a/src/sempy_labs/_deployment_pipelines.py b/src/sempy_labs/_deployment_pipelines.py
index 4ff8b94f..b8089131 100644
--- a/src/sempy_labs/_deployment_pipelines.py
+++ b/src/sempy_labs/_deployment_pipelines.py
@@ -148,7 +148,7 @@ def list_deployment_pipeline_stage_items(
         raise ValueError(
             f"{icons.red_dot} The '{stage_name}' stage does not exist within the '{deployment_pipeline}' deployment pipeline."
         )
-    stage_id = dfPS_filt["Deployment Pipeline Stage ID"].iloc[0]
+    stage_id = dfPS_filt["Deployment Pipeline Stage Id"].iloc[0]
 
     client = fabric.FabricRestClient()
     response = client.get(
diff --git a/src/sempy_labs/_helper_functions.py b/src/sempy_labs/_helper_functions.py
index 64b33f45..5589cef9 100644
--- a/src/sempy_labs/_helper_functions.py
+++ b/src/sempy_labs/_helper_functions.py
@@ -975,7 +975,7 @@ def resolve_deployment_pipeline_id(deployment_pipeline: str) -> UUID:
 
     Returns
     -------
-    UUID
+    uuid.UUID
         The deployment pipeline Id.
     """