diff --git a/tests/api.c b/tests/api.c
index 8f561b0648..fbe9cae3d4 100644
--- a/tests/api.c
+++ b/tests/api.c
@@ -1444,7 +1444,7 @@ static int test_dual_alg_crit_ext_support(void)
     XFREE(root, NULL, DYNAMIC_TYPE_TMP_BUFFER);
     XFREE(server, NULL, DYNAMIC_TYPE_TMP_BUFFER);
 
-    free(serverKey);
+    XFREE(serverKey, NULL, DYNAMIC_TYPE_TMP_BUFFER);
 
     return EXPECT_RESULT();
 }
@@ -1519,7 +1519,7 @@ static int test_dual_alg_support(void)
     XFREE(root, NULL, DYNAMIC_TYPE_TMP_BUFFER);
     XFREE(server, NULL, DYNAMIC_TYPE_TMP_BUFFER);
 
-    free(serverKey);
+    XFREE(serverKey, NULL, DYNAMIC_TYPE_TMP_BUFFER);
 
     return EXPECT_RESULT();
 }
diff --git a/tests/api/test_pkcs7.c b/tests/api/test_pkcs7.c
index 3cfb22e8ea..0425d64373 100644
--- a/tests/api/test_pkcs7.c
+++ b/tests/api/test_pkcs7.c
@@ -4718,7 +4718,7 @@ int test_wc_PKCS7_DecodeCompressedData(void)
     XFREE(decompressed, heap, DYNAMIC_TYPE_TMP_BUFFER);
 
     if (cert_buf != NULL)
-        free(cert_buf);
+        XFREE(cert_buf, NULL, DYNAMIC_TYPE_TMP_BUFFER);
     wc_PKCS7_Free(pkcs7);
 #endif
     return EXPECT_RESULT();