3 files changed, 6 insertions, 3 deletions
diff --git a/crypto_aead/lilliputaei128v1/ref/lilliput-ae-i.c b/crypto_aead/lilliputaei128v1/ref/lilliput-ae-i.c
index 945c75c..82dd0b7 100644
--- a/crypto_aead/lilliputaei128v1/ref/lilliput-ae-i.c
+++ b/crypto_aead/lilliputaei128v1/ref/lilliput-ae-i.c
@@ -7,6 +7,7 @@
 void lilliput_ae_encrypt(
     size_t         message_len,   const uint8_t message[message_len],
     size_t         auth_data_len, const uint8_t auth_data[auth_data_len],
+    const uint8_t key[KEY_BYTES],
     const uint8_t  nonce[NONCE_BYTES],
 
     size_t *ciphertext_len, uint8_t ciphertext[message_len+BLOCK_BYTES],
@@ -18,6 +19,7 @@ void lilliput_ae_encrypt(
 bool lilliput_ae_decrypt(
     size_t ciphertext_len, const uint8_t ciphertext[ciphertext_len],
     size_t auth_data_len, const uint8_t auth_data[auth_data_len],
+    const uint8_t key[KEY_BYTES],
     const uint8_t nonce[NONCE_BYTES],
     const uint8_t tag[TAG_BYTES],
 
diff --git a/crypto_aead/lilliputaei128v1/ref/lilliput-ae.h b/crypto_aead/lilliputaei128v1/ref/lilliput-ae.h
index 973533c..d5c19b3 100644
--- a/crypto_aead/lilliputaei128v1/ref/lilliput-ae.h
+++ b/crypto_aead/lilliputaei128v1/ref/lilliput-ae.h
@@ -11,6 +11,7 @@
 void lilliput_ae_encrypt(
     size_t message_len,   const uint8_t message[message_len],
     size_t auth_data_len, const uint8_t auth_data[auth_data_len],
+    const uint8_t key[KEY_BYTES],
     const uint8_t nonce[NONCE_BYTES],
 
     size_t *ciphertext_len, uint8_t ciphertext[message_len+BLOCK_BYTES],
@@ -20,6 +21,7 @@ void lilliput_ae_encrypt(
 bool lilliput_ae_decrypt(
     size_t ciphertext_len, const uint8_t ciphertext[ciphertext_len],
     size_t auth_data_len,  const uint8_t auth_data[auth_data_len],
+    const uint8_t key[KEY_BYTES],
     const uint8_t nonce[NONCE_BYTES],
     const uint8_t tag[TAG_BYTES],
 
diff --git a/crypto_aead/lilliputaei128v1/ref/test/test-ae-roundtrip.c b/crypto_aead/lilliputaei128v1/ref/test/test-ae-roundtrip.c
index d97a813..f1cb24c 100644
--- a/crypto_aead/lilliputaei128v1/ref/test/test-ae-roundtrip.c
+++ b/crypto_aead/lilliputaei128v1/ref/test/test-ae-roundtrip.c
@@ -87,7 +87,7 @@ int main()
         lilliput_ae_encrypt(
             v->message_len, v->message,
             v->auth_len, v->auth,
-            v->nonce,
+            v->nonce, v->key,
             &ciphertext_len, ciphertext,
             tag
         );
@@ -97,8 +97,7 @@ int main()
         bool valid = lilliput_ae_decrypt(
             ciphertext_len, ciphertext,
             v->auth_len, v->auth,
-            v->nonce,
-            tag,
+            v->nonce, v->key, tag,
             &deciphered_len, deciphered
         );