Update tests for AES library