From a97ef0ecd577357a2108632008bf7bdace5fb13d Mon Sep 17 00:00:00 2001
From: vatsrahul1001 <43964496+vatsrahul1001@users.noreply.github.com>
Date: Fri, 12 Apr 2024 08:38:49 +0530
Subject: [PATCH] Fix task name to get cluster IP using xcom (#1507)

We were not using correct task for cluster IP retrieval. Updated task name correctly.
---
 astronomer/providers/sftp/example_dags/example_sftp.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/astronomer/providers/sftp/example_dags/example_sftp.py b/astronomer/providers/sftp/example_dags/example_sftp.py
index 9af9eedcb..8c2426b7a 100644
--- a/astronomer/providers/sftp/example_dags/example_sftp.py
+++ b/astronomer/providers/sftp/example_dags/example_sftp.py
@@ -167,7 +167,7 @@ def revoke_inbound_rules(task_instance: TaskInstance) -> None:
         FromPort=22,
         ToPort=22,
         GroupId=task_instance.xcom_pull(
-            key="cluster_response_master_security_group", task_ids=["describe_created_cluster"]
+            key="instance_response_master_security_group", task_ids=["create_ec2_instance"]
         )[0],
         IpProtocol="tcp",
     )