diff --git a/programs/ssl/ssl_client2.c b/programs/ssl/ssl_client2.c
index 4235913f0..415caa0cb 100644
--- a/programs/ssl/ssl_client2.c
+++ b/programs/ssl/ssl_client2.c
@@ -2371,8 +2371,11 @@ send_request:
         opt.serialize = 0;
         mbedtls_printf( " Serializing live connection..." );
 
-        if( ( ret = mbedtls_ssl_context_save( &ssl, NULL, 0, &len ) )
-                != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL )
+        ret = mbedtls_ssl_context_save( &ssl, NULL, 0, &len );
+
+        /* Allow stub implementation returning 0 for now */
+        if( ret != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL &&
+            ret != 0 )
         {
             mbedtls_printf( " failed\n  ! mbedtls_ssl_context_save returned "
                             "-0x%x\n\n", -ret );
diff --git a/programs/ssl/ssl_server2.c b/programs/ssl/ssl_server2.c
index 7b2c18453..6103688b2 100644
--- a/programs/ssl/ssl_server2.c
+++ b/programs/ssl/ssl_server2.c
@@ -3356,8 +3356,11 @@ data_exchange:
         opt.serialize = 0;
         mbedtls_printf( " Serializing live connection..." );
 
-        if( ( ret = mbedtls_ssl_context_save( &ssl, NULL, 0, &len ) )
-                != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL )
+        ret = mbedtls_ssl_context_save( &ssl, NULL, 0, &len );
+
+        /* Allow stub implementation returning 0 for now */
+        if( ret != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL &&
+            ret != 0 )
         {
             mbedtls_printf( " failed\n  ! mbedtls_ssl_context_save returned "
                             "-0x%x\n\n", -ret );