Updated the tests for aws api