diff --git a/Android.gralloc.mk b/Android.gralloc.mk
index 4a0b125..ff269c0 100644
--- a/Android.gralloc.mk
+++ b/Android.gralloc.mk
@@ -5,6 +5,7 @@
 LOCAL_CPP_EXTENSION := .cc
 
 LOCAL_SRC_FILES += \
-	cros_gralloc/cros_alloc_device.cc \
+	cros_gralloc/cros_gralloc_buffer.cc \
+	cros_gralloc/cros_gralloc_driver.cc \
 	cros_gralloc/cros_gralloc_helpers.cc \
-	cros_gralloc/cros_gralloc_module.cc
+	cros_gralloc/gralloc0/gralloc0.cc