AES: unit test updates