diff --git a/src/api/pmi.c b/src/api/pmi.c
index b64d81af2710cc85f0012e25c3f6bac3c3b9cc28..6d89b8c805f395e665c3c50afd53723fd282114a 100644
--- a/src/api/pmi.c
+++ b/src/api/pmi.c
@@ -78,6 +78,7 @@
 #include <slurm/slurm_errno.h>
 
 #include "src/api/slurm_pmi.h"
+#include "src/common/malloc.h"
 
 #define KVS_STATE_LOCAL    0
 #define KVS_STATE_DEFUNCT  1