diff --git a/src/ssl.c b/src/ssl.c
index 17a1c3f10..f4fe18be6 100644
--- a/src/ssl.c
+++ b/src/ssl.c
@@ -26397,14 +26397,15 @@ int wolfSSL_CTX_set_msg_callback_arg(WOLFSSL_CTX *ctx, void* arg)
     (void)arg;
     return WOLFSSL_FAILURE;
 }
+#endif
+
 int wolfSSL_set_msg_callback_arg(WOLFSSL *ssl, void* arg)
 {
-    WOLFSSL_STUB("SSL_set_msg_callback_arg");
-    (void)ssl;
-    (void)arg;
-    return WOLFSSL_FAILURE;
+    WOLFSSL_ENTER("wolfSSL_set_msg_callback_arg");
+    ssl->protoMsgCtx = arg;
+    return SSL_FAILURE;
 }
-#endif
+
 #endif
 
 #endif /* WOLFCRYPT_ONLY */