diff --git a/tests/suites/main_test.function b/tests/suites/main_test.function
index 9ac5a4a..f34c1bd 100644
--- a/tests/suites/main_test.function
+++ b/tests/suites/main_test.function
@@ -239,9 +239,13 @@
 {
 #if defined(MBEDTLS_TEST_HOOKS) && defined (MBEDTLS_ERROR_C)
     mbedtls_test_hook_error_add = &mbedtls_test_err_add_check;
-    mbedtls_test_hook_value = &mbedtls_test_hook_value_check;
 #endif
 
+#if defined(MBEDTLS_TEST_HOOKS)
+    mbedtls_test_hook_assert_test = &mbedtls_test_assert_test;
+#endif
+
+
     int ret = mbedtls_test_platform_setup();
     if( ret != 0 )
     {