diff --git a/tests/suites/test_suite_aes.function b/tests/suites/test_suite_aes.function
index 24e8f7a..f1e9033 100644
--- a/tests/suites/test_suite_aes.function
+++ b/tests/suites/test_suite_aes.function
@@ -305,11 +305,11 @@
     unsigned char* src_str_next;
     int key_len;
 
-    memset(key_str, 0x00, 32);
-    memset(iv_str, 0x00, 16);
-    memset(src_str, 0x00, 64);
-    memset(dst_str, 0x00, 64);
-    memset(output, 0x00, 32);
+    memset( key_str, 0x00, sizeof( key_str ) );
+    memset( iv_str, 0x00, sizeof( iv_str ) );
+    memset( src_str, 0x00, sizeof( src_str ) );
+    memset( dst_str, 0x00, sizeof( dst_str ) );
+    memset( output, 0x00, sizeof( output ) );
     mbedtls_aes_init( &ctx );
 
     TEST_ASSERT( strlen( hex_key_string ) <= ( 32 * 2 ) );