1 files changed, 4 insertions, 1 deletions
diff --git a/test/aes128ecb_encrypt_block.c b/test/aes128ecb_encrypt_block.c
index 27c7a0e..8f0bf0a 100644
--- a/test/aes128ecb_encrypt_block.c
+++ b/test/aes128ecb_encrypt_block.c
@@ -20,6 +20,7 @@ static void exit_with_usage(const char* argv0)
 int main(int argc, char** argv)
 {
     __declspec(align(16)) AesBlock128 plain, key, cipher;
+    __declspec(align(16)) Aes128KeySchedule key_schedule;
 
     if (argc < 2)
         exit_with_usage(argv[0]);
@@ -30,6 +31,8 @@ int main(int argc, char** argv)
         exit_with_usage(argv[0]);
     }
 
+    aes128_expand_key_schedule(key, &key_schedule);
+
     for (int i = 2; i < argc; ++i)
     {
         if (parse_aes_block128(&plain, argv[i]) != 0)
@@ -37,7 +40,7 @@ int main(int argc, char** argv)
             fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
             continue;
         }
-        cipher = aes128ecb_encrypt(plain, key);
+        cipher = aes128ecb_encrypt(plain, &key_schedule);
         print_aes_block128(&cipher);
     }