1 files changed, 3 insertions, 4 deletions
diff --git a/aes.h b/aes.h
index 9c5d786..1c8af6c 100644
--- a/aes.h
+++ b/aes.h
@@ -18,13 +18,12 @@
   #define ECB 1
 #endif
 
-#define AES128
-
+#define AES256
 
 #if defined(ECB) && ECB
 
-void AES_ECB_encrypt(const uint8_t* input, const uint8_t* key, uint8_t *output);
-void AES_ECB_decrypt(const uint8_t* input, const uint8_t* key, uint8_t *output);
+void AES_ECB_encrypt(const uint8_t* input, const uint8_t* key, uint8_t *output, const uint32_t length);
+void AES_ECB_decrypt(const uint8_t* input, const uint8_t* key, uint8_t *output, const uint32_t length);
 
 #endif // #if defined(ECB) && ECB