From 0aa0f2628922446390bf13cfc5c7b62317ed0b67 Mon Sep 17 00:00:00 2001
From: Daniel Pouzzner <douzzer@wolfssl.com>
Date: Fri, 19 Jul 2024 13:57:13 -0500
Subject: [PATCH] wolfcrypt/src/dilithium.c: fix null deref in
 wc_dilithium_init_ex().

---
 wolfcrypt/src/dilithium.c | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/wolfcrypt/src/dilithium.c b/wolfcrypt/src/dilithium.c
index 625d6bf29..bd5b10272 100644
--- a/wolfcrypt/src/dilithium.c
+++ b/wolfcrypt/src/dilithium.c
@@ -6952,10 +6952,9 @@ int wc_dilithium_init_ex(dilithium_key* key, void* heap, int devId)
         key->idLen = 0;
         key->labelLen = 0;
     #endif
+        key->heap = heap;
     }
 
-    key->heap = heap;
-
     return ret;
 }