code to get tests working for aws