diff --git a/yarrow256.c b/yarrow256.c
index 4910049aa77eabfb04c064c1ff455fe3ce6755ad..6a275d69a6defc77b0a2f9ff7b1d311f7c44c7ef 100644
--- a/yarrow256.c
+++ b/yarrow256.c
@@ -195,7 +195,7 @@ yarrow_fast_reseed(struct yarrow256_ctx *ctx)
   /* Iterate */
   yarrow_iterate(digest);
 
-  aes_set_key(&ctx->key, sizeof(digest), digest);
+  aes_set_encrypt_key(&ctx->key, sizeof(digest), digest);
 
   /* Derive new counter value */
   memset(ctx->counter, 0, sizeof(ctx->counter));
@@ -332,7 +332,7 @@ yarrow_gate(struct yarrow256_ctx *ctx)
   for (i = 0; i < sizeof(key); i+= AES_BLOCK_SIZE)
     yarrow_generate_block(ctx, key + i);
 
-  aes_set_key(&ctx->key, sizeof(key), key);
+  aes_set_encrypt_key(&ctx->key, sizeof(key), key);
 }
 
 void