diff --git a/src/sig_stfl/lms/CMakeLists.txt b/src/sig_stfl/lms/CMakeLists.txt
index e47452eb5..b04030e83 100644
--- a/src/sig_stfl/lms/CMakeLists.txt
+++ b/src/sig_stfl/lms/CMakeLists.txt
@@ -37,6 +37,7 @@ set(SRCS
 
 
 add_library(lms OBJECT ${SRCS})
+target_include_directories(lms PRIVATE ${LIBOQS_ROOT_DIR}/include)
 set(_LMS_OBJS ${_LMS_OBJS} $<TARGET_OBJECTS:lms>)
 set(LMS_OBJS ${_LMS_OBJS} PARENT_SCOPE)
 
diff --git a/src/sig_stfl/lms/external/common_defs.h b/src/sig_stfl/lms/external/common_defs.h
index 1c7c85d38..2a6473810 100644
--- a/src/sig_stfl/lms/external/common_defs.h
+++ b/src/sig_stfl/lms/external/common_defs.h
@@ -9,7 +9,7 @@
 
 #include <stdint.h>
 #include <stdbool.h>
-
+#include <oqs/oqs.h>
 #define MAX_HASH   32 /* Length of the largest hash we support */
 
 /* The I (Merkle tree identifier) value is 16 bytes long */
diff --git a/src/sig_stfl/xmss/external/wots.c b/src/sig_stfl/xmss/external/wots.c
index 6ba8dd9f9..3d87c206b 100644
--- a/src/sig_stfl/xmss/external/wots.c
+++ b/src/sig_stfl/xmss/external/wots.c
@@ -187,7 +187,7 @@ void wots_pk_from_sig(const xmss_params *params, unsigned char *pk,
                       const unsigned char *sig, const unsigned char *msg,
                       const unsigned char *pub_seed, uint32_t addr[8])
 {
-    unsigned int *lengths = OQS_MEM_malloc(params->wots_len, sizeof(unsigned int ));
+    unsigned int *lengths = OQS_MEM_calloc(params->wots_len, sizeof(unsigned int ));
     const size_t thash_buf_len = 2 * params->padding_len + 4 * params->n + 32;
     unsigned char *thash_buf = OQS_MEM_malloc(thash_buf_len);
     unsigned int i;