diff --git a/drivers/dma-buf/dma-heap.c b/drivers/dma-buf/dma-heap.c
index 104430200586..c9077f645b8d 100644
--- a/drivers/dma-buf/dma-heap.c
+++ b/drivers/dma-buf/dma-heap.c
@@ -105,8 +105,8 @@ struct dma_buf *dma_heap_buffer_alloc(struct dma_heap *heap, size_t len,
 EXPORT_SYMBOL_GPL(dma_heap_buffer_alloc);
 
 int dma_heap_bufferfd_alloc(struct dma_heap *heap, size_t len,
-			    unsigned int fd_flags,
-			    unsigned int heap_flags)
+			    u32 fd_flags,
+			    u64 heap_flags)
 {
 	struct dma_buf *dmabuf;
 	int fd;
diff --git a/include/linux/dma-heap.h b/include/linux/dma-heap.h
index 15bd3a7cd24d..90238183d53f 100644
--- a/include/linux/dma-heap.h
+++ b/include/linux/dma-heap.h
@@ -107,8 +107,8 @@ void dma_heap_buffer_free(struct dma_buf *);
  * @heap_flags:	flags to pass to the dma heap
  */
 int dma_heap_bufferfd_alloc(struct dma_heap *heap, size_t len,
-			    unsigned int fd_flags,
-			    unsigned int heap_flags);
+			    u32 fd_flags,
+			    u64 heap_flags);
 
 /**
  * dma_heap_try_get_pool_size_kb - Returns total dma-heap pool size in kb