From b615309a7b9fa39933bc33d5e5f79a6d1cf8cba8 Mon Sep 17 00:00:00 2001
From: John Safranek <john@wolfssl.com>
Date: Thu, 10 Jun 2021 15:35:27 -0700
Subject: [PATCH] update FFDHE4096 test with the updated usage

---
 wolfcrypt/test/test.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/wolfcrypt/test/test.c b/wolfcrypt/test/test.c
index 459f88c99..ec6b4fef6 100644
--- a/wolfcrypt/test/test.c
+++ b/wolfcrypt/test/test.c
@@ -16844,9 +16844,13 @@ WOLFSSL_TEST_SUBROUTINE int dh_test(void)
         ERROR_OUT(-8127, done);
     #endif
     #ifdef HAVE_FFDHE_4096
+    #ifdef HAVE_PUBLIC_FFDHE
     ret = dh_ffdhe_test(&rng, wc_Dh_ffdhe4096_Get());
     if (ret != 0)
         ERROR_OUT(-8128, done);
+    #else
+    ret = dh_ffdhe_test(&rng, WC_FFDHE_4096);
+    #endif
     #endif
 #endif /* !WC_NO_RNG */
 #endif /* HAVE_FIPS_VERSION == 2 && !WOLFSSL_SP_ARM64_ASM */