diff --git a/test/lib/nvme/unit/nvme_impl.h b/test/lib/nvme/unit/nvme_impl.h
index 9b9ff4714..0e0941806 100644
--- a/test/lib/nvme/unit/nvme_impl.h
+++ b/test/lib/nvme/unit/nvme_impl.h
@@ -35,10 +35,18 @@
 #define __NVME_IMPL_H__
 
 #include <stdio.h>
+#include <stdlib.h>
 #include <stdint.h>
 #include <pthread.h>
 
-#define nvme_malloc(tag, size, align, phys_addr)	malloc(size)
+static inline void *
+nvme_malloc(const char *tag, size_t size, unsigned align, uint64_t *phys_addr)
+{
+	void *buf = calloc(1, size);
+	*phys_addr = (uint64_t)buf;
+	return buf;
+}
+
 #define nvme_free(buf)			free(buf)
 #define OUTBUF_SIZE 1024
 extern char outbuf[OUTBUF_SIZE];