diff --git a/tests/suites/main_test.function b/tests/suites/main_test.function
index f34c1bd..3044bed 100644
--- a/tests/suites/main_test.function
+++ b/tests/suites/main_test.function
@@ -242,10 +242,9 @@
 #endif
 
 #if defined(MBEDTLS_TEST_HOOKS)
-    mbedtls_test_hook_assert_test = &mbedtls_test_assert_test;
+    mbedtls_test_hook_test_fail = &mbedtls_test_fail;
 #endif
 
-
     int ret = mbedtls_test_platform_setup();
     if( ret != 0 )
     {