Index: mojo/examples/aura_demo/aura_demo.cc |
diff --git a/mojo/examples/aura_demo/aura_demo.cc b/mojo/examples/aura_demo/aura_demo.cc |
index 4ed3bd764c32137cba7d3e89d88014a0b87a06b1..2ac77cd5dc350918db84703c94160958d3dc8d57 100644 |
--- a/mojo/examples/aura_demo/aura_demo.cc |
+++ b/mojo/examples/aura_demo/aura_demo.cc |
@@ -116,12 +116,10 @@ class DemoWindowTreeClient : public aura::client::WindowTreeClient { |
DISALLOW_COPY_AND_ASSIGN(DemoWindowTreeClient); |
}; |
-class AuraDemo : public ShellClientStub { |
+class AuraDemo : public ShellClient { |
public: |
explicit AuraDemo(ScopedMessagePipeHandle shell_handle) |
- : shell_(shell_handle.Pass()) { |
- shell_.SetPeer(this); |
- |
+ : shell_(shell_handle.Pass(), this) { |
screen_.reset(DemoScreen::Create()); |
gfx::Screen::SetScreenInstance(gfx::SCREEN_TYPE_NATIVE, screen_.get()); |