@@ -755,7 +755,7 @@ def execute_athena_query_async(
         # return athena_data_client.execute_statement(
         return athena_data_client.start_query_execution(
             QueryString=query,
-            QueryExecutionContext={"Database": database},
+            QueryExecutionContext={"Database": database, "Catalog": data_source},
             WorkGroup=workgroup,
         )
Original file line number	Diff line number	Diff line change
`@@ -755,7 +755,7 @@ def execute_athena_query_async(`
`755`	`755`	`# return athena_data_client.execute_statement(`
`756`	`756`	`return athena_data_client.start_query_execution(`
`757`	`757`	`QueryString=query,`
`758`		`- QueryExecutionContext={"Database": database},`
	`758`	`+ QueryExecutionContext={"Database": database, "Catalog": data_source},`
`759`	`759`	`WorkGroup=workgroup,`
`760`	`760`	`)`
`761`	`761`