1 files changed, 8 insertions, 4 deletions
diff --git a/crypto_aead/lilliputaei128v1/ref/test/tweakey.c b/crypto_aead/lilliputaei128v1/ref/test/tweakey.c
index ddf595b..1f5aacf 100644
--- a/crypto_aead/lilliputaei128v1/ref/test/tweakey.c
+++ b/crypto_aead/lilliputaei128v1/ref/test/tweakey.c
@@ -8,6 +8,9 @@
 #define ARRAY_END(A) (A+ARRAY_NB(A))
 
 
+#define ROUNDS 32
+
+
 struct vector_input
 {
     char * name;
@@ -96,13 +99,14 @@ int main()
         fprintf(dump, "Tweakey Schedule\n");
 
         uint8_t rtk[ROUND_TWEAKEY_BYTES];
-        for (uint8_t i=0; i<32; i++)
+        tweakey_state_extract(&tk, rtk, 0);
+
+        for (uint8_t i=1; i<ROUNDS; i++)
         {
-            tweakey_state_extract(&tk, rtk, i);
-            fprintf(dump, "     Round Tweakey  %"PRIu8"\n", i+1);
+            fprintf(dump, "     Round Tweakey  %"PRIu8"\n", i);
             tweakey_state_update(&tk);
+            tweakey_state_extract(&tk, rtk, i);
         }
-        tweakey_state_extract(&tk, rtk, 32);
 
         fclose(dump);
     }