fix up tests for aws