diff --git a/system/vulkan_enc/AndroidHardwareBuffer.h b/system/vulkan_enc/AndroidHardwareBuffer.h
index cab626b..c5e64ee 100644
--- a/system/vulkan_enc/AndroidHardwareBuffer.h
+++ b/system/vulkan_enc/AndroidHardwareBuffer.h
@@ -32,8 +32,7 @@
     const VkImageCreateFlags vk_create,
     const VkImageUsageFlags vk_usage);
 
-void updateMemoryTypeBitsForAndroidHardwareBuffers(
-    uint32_t* memoryTypeBits);
+void updateMemoryTypeBits(uint32_t* memoryTypeBits, uint32_t colorBufferMemoryIndex);
 
 VkResult getAndroidHardwareBufferPropertiesANDROID(
     Gralloc* grallocHelper,