diff --git a/tests/suites/test_suite_aes.function b/tests/suites/test_suite_aes.function
index f61f71c..1832f2d 100644
--- a/tests/suites/test_suite_aes.function
+++ b/tests/suites/test_suite_aes.function
@@ -394,13 +394,15 @@
 
     TEST_INVALID_PARAM_RET( MBEDTLS_ERR_AES_BAD_INPUT_DATA,
                             mbedtls_aes_setkey_dec( &dummy_ctx, NULL, 128 ) );
+}
+/* END_CASE */
 
+/* BEGIN_CASE */
+void aes_valid_param( )
+{
     /* These calls accept NULL */
     TEST_VALID_PARAM( mbedtls_aes_free( NULL ) );
     TEST_VALID_PARAM( mbedtls_aes_xts_free( NULL ) );
-
-exit:
-    return;
 }
 /* END_CASE */
 
diff --git a/tests/suites/test_suite_aes.rest.data b/tests/suites/test_suite_aes.rest.data
index 14b78a6..a5d843d 100644
--- a/tests/suites/test_suite_aes.rest.data
+++ b/tests/suites/test_suite_aes.rest.data
@@ -13,6 +13,9 @@
 AES - Invalid parameters
 aes_invalid_param:
 
+AES - Valid parameters
+aes_valid_param:
+
 AES Selftest
 depends_on:MBEDTLS_SELF_TEST
 aes_selftest: