diff --git a/tests/suites/test_suite_aes.function b/tests/suites/test_suite_aes.function
index 7b1306a..9118a98 100644
--- a/tests/suites/test_suite_aes.function
+++ b/tests/suites/test_suite_aes.function
@@ -159,7 +159,7 @@
     mbedtls_test_rnd_std_rand(NULL, key, sizeof(key));
 
     // random input
-    mbedtls_test_rnd_std_rand(NULL, input, sizeof(input));
+    mbedtls_test_rnd_std_rand(NULL, input, length);
 
 
     // complete encryption in one call