diff --git a/tests/suites/helpers.function b/tests/suites/helpers.function
index 2eff043..6d4438d 100644
--- a/tests/suites/helpers.function
+++ b/tests/suites/helpers.function
@@ -13,6 +13,8 @@
 #define mbedtls_fprintf    fprintf
 #define mbedtls_printf     printf
 #define mbedtls_snprintf   snprintf
+#define MBEDTLS_EXIT_SUCCESS EXIT_SUCCESS
+#define MBEDTLS_EXIT_FAILURE EXIT_FAILURE
 #endif
 
 #if defined(MBEDTLS_MEMORY_BUFFER_ALLOC_C)
diff --git a/tests/suites/main_test.function b/tests/suites/main_test.function
index 525df5b..c2e3f6b 100644
--- a/tests/suites/main_test.function
+++ b/tests/suites/main_test.function
@@ -356,7 +356,7 @@
                         if(  unmet_dependencies[ i-1 ] == NULL )
                         {
                             mbedtls_printf("FATAL: Out of memory\n");
-                            mbedtls_exit( MBEDTLS_PLATFORM_STD_EXIT_FAILURE );
+                            mbedtls_exit( MBEDTLS_EXIT_FAILURE );
                         }
                         unmet_dep_count++;
                     }