diff --git a/packages/aws/tests/python/test_redshift_data.py b/packages/aws/tests/python/test_redshift_data.py index 9a64740ebe..f5217dfa92 100644 --- a/packages/aws/tests/python/test_redshift_data.py +++ b/packages/aws/tests/python/test_redshift_data.py @@ -99,4 +99,4 @@ def test_execute_statement(redshift): tables = aws.execute_redshift_statement(TEST_SQL) print("RETURNED TABLE:") print(tables) - assert tables.get(0)["City"] == "Vancouver" + assert tables[0]["City"] == "Vancouver"