diff --git a/src/VirtualDisplay.h b/src/VirtualDisplay.h
index cd2356e..55b2166 100644
--- a/src/VirtualDisplay.h
+++ b/src/VirtualDisplay.h
@@ -23,8 +23,7 @@
 #include <gui/CpuConsumer.h>
 #include <gui/IGraphicBufferProducer.h>
 
-#include <ui/DisplayConfig.h>
-#include <ui/DisplayInfo.h>
+#include <ui/DisplayMode.h>
 #include <ui/DisplayState.h>
 #include <ui/Rect.h>
 
@@ -34,7 +33,7 @@
 
 class VirtualDisplay : public RefBase {
   public:
-    VirtualDisplay(DisplayConfig *config, ui::DisplayState* state,
+    VirtualDisplay(ui::DisplayMode *mode, ui::DisplayState* state,
                    uint32_t width, uint32_t height,
                    sp<CpuConsumer::FrameAvailableListener> listener);