diff --git a/kernel/ethosu_buffer.h b/kernel/ethosu_buffer.h
index 1829fbe..8bef2d2 100644
--- a/kernel/ethosu_buffer.h
+++ b/kernel/ethosu_buffer.h
@@ -31,25 +31,22 @@
  * Types
  ****************************************************************************/
 
+struct ethosu_dma_mem;
 struct ethosu_device;
 struct device;
 
 /**
- * struct ethosu_buffer - Buffer
- * @dev:		Device
- * @file:		File
- * @kref:		Reference counting
- * @size:		Size of the buffer
- * @cpu_addr:		Kernel mapped address
- * @dma_addr:		DMA address
+ * struct ethosu_buffer - User data buffer
+ * @dev:	Device
+ * @file:	File
+ * @kref:	Reference counting
+ * @dma_mem:	DMA memory allocated for the buffer
  */
 struct ethosu_buffer {
-	struct device *dev;
-	struct file   *file;
-	struct kref   kref;
-	size_t        size;
-	void          *cpu_addr;
-	dma_addr_t    dma_addr;
+	struct device         *dev;
+	struct file           *file;
+	struct kref           kref;
+	struct ethosu_dma_mem *dma_mem;
 };
 
 /****************************************************************************