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 */