From 7301f73d40a4a8825f4f723db4fcf829764f2738 Mon Sep 17 00:00:00 2001
From: Aimery Methena <159072740+aimethed@users.noreply.github.com>
Date: Mon, 11 Nov 2024 15:54:44 -0500
Subject: [PATCH] add ImageConfig back to postgres connection yaml (#2386)

---
 athena-postgresql/athena-postgresql-connection.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/athena-postgresql/athena-postgresql-connection.yaml b/athena-postgresql/athena-postgresql-connection.yaml
index cebadb96ec..79b3cb38c8 100644
--- a/athena-postgresql/athena-postgresql-connection.yaml
+++ b/athena-postgresql/athena-postgresql-connection.yaml
@@ -69,6 +69,8 @@ Resources:
       ImageUri: !Sub
         - '${Account}.dkr.ecr.${AWS::Region}.amazonaws.com/athena-federation-repository-postgresql:2022.47.1'
         - Account: !If [IsRegionBAH, 084828588479, !If [IsRegionHKG, 183295418215, 292517598671]]
+      ImageConfig:
+        Command: [ !Sub "com.amazonaws.athena.connectors.postgresql.PostGreSqlCompositeHandler" ]
       Description: "Enables Amazon Athena to communicate with PostgreSQL using JDBC"
       Timeout: 900
       MemorySize: 3008