Fix xcom key in sftp revoke inbound rule task

astronomer · Apr 12, 2024 · 678bb5e · 678bb5e
1 parent 23eb03f
commit 678bb5e
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/astronomer/providers/sftp/example_dags/example_sftp.py b/astronomer/providers/sftp/example_dags/example_sftp.py
@@ -166,9 +166,7 @@ def revoke_inbound_rules(task_instance: TaskInstance) -> None:
         CidrIp=ip_range,
         FromPort=22,
         ToPort=22,
-        GroupId=task_instance.xcom_pull(
-            key="instance_response_master_security_group", task_ids=["create_ec2_instance"]
-        )[0],
+        GroupId=task_instance.xcom_pull(key=INSTANCE_SECURITY_GROUP, task_ids=["create_ec2_instance"])[0],
         IpProtocol="tcp",
     )
     logging.info("%s", response)