diff --git a/source/common/unified_malloc_framework/include/umf/memory_pool_ops.h b/source/common/unified_malloc_framework/include/umf/memory_pool_ops.h
index 16a7d1252e..03660f695c 100644
--- a/source/common/unified_malloc_framework/include/umf/memory_pool_ops.h
+++ b/source/common/unified_malloc_framework/include/umf/memory_pool_ops.h
@@ -12,6 +12,7 @@
 #define UMF_MEMORY_POOL_OPS_H 1
 
 #include <umf/base.h>
+#include <umf/memory_provider.h>
 
 #ifdef __cplusplus
 extern "C" {