diff --git a/programs/psa/crypto_examples.c b/programs/psa/crypto_examples.c
index d7a667b..9947a70 100644
--- a/programs/psa/crypto_examples.c
+++ b/programs/psa/crypto_examples.c
@@ -5,7 +5,10 @@
 #include "mbedtls/platform.h"
 #else
 #include <stdio.h>
+#include <stdlib.h>
+#define MBEDTLS_EXIT_FAILURE EXIT_FAILURE
 #define mbedtls_printf printf
+#define mbedtls_exit exit
 #endif
 
 #define ASSERT( predicate )                                                   \
diff --git a/programs/psa/key_ladder_demo.c b/programs/psa/key_ladder_demo.c
index 0943bf5..c9d7676 100644
--- a/programs/psa/key_ladder_demo.c
+++ b/programs/psa/key_ladder_demo.c
@@ -66,6 +66,7 @@
 #define mbedtls_calloc       calloc
 #define mbedtls_free         free
 #define mbedtls_printf       printf
+#define mbedtls_exit         exit
 #endif
 #include <stdio.h>
 #include <string.h>