AES: more unit test fixes