diff --git a/src/AndroidPixelBuffer.h b/src/AndroidPixelBuffer.h
index 174665f..5d06d8e 100644
--- a/src/AndroidPixelBuffer.h
+++ b/src/AndroidPixelBuffer.h
@@ -21,8 +21,7 @@
 #include <utils/Mutex.h>
 #include <utils/RefBase.h>
 
-#include <ui/DisplayConfig.h>
-#include <ui/DisplayInfo.h>
+#include <ui/DisplayMode.h>
 #include <ui/DisplayState.h>
 #include <ui/Rect.h>
 
@@ -37,7 +36,7 @@
   public:
     AndroidPixelBuffer();
 
-    virtual void setDisplayInfo(DisplayConfig* config, ui::DisplayState* state);
+    virtual void setDisplayInfo(ui::DisplayMode* mode, ui::DisplayState* state);
 
     virtual void setWindowSize(uint32_t width, uint32_t height);