1 files changed, 5 insertions, 7 deletions
diff --git a/include/cru/ui/controls/RootControl.h b/include/cru/ui/controls/RootControl.h
index 41424286..a1878836 100644
--- a/include/cru/ui/controls/RootControl.h
+++ b/include/cru/ui/controls/RootControl.h
@@ -1,16 +1,14 @@
 #pragma once
 #include "LayoutControl.h"
 
-#include "cru/common/Base.h"
 #include "cru/common/Event.h"
-#include "cru/platform/gui/Base.h"
 #include "cru/platform/gui/Window.h"
-#include "cru/ui/Base.h"
 #include "cru/ui/host/WindowHost.h"
 #include "cru/ui/render/StackLayoutRenderObject.h"
 
 namespace cru::ui::controls {
-class CRU_UI_API RootControl : public LayoutControl {
+class CRU_UI_API RootControl
+    : public LayoutControl<render::StackLayoutRenderObject> {
  protected:
   explicit RootControl(Control* attached_control);
 
@@ -20,7 +18,9 @@ class CRU_UI_API RootControl : public LayoutControl {
   ~RootControl() override;
 
  public:
-  render::RenderObject* GetRenderObject() const override;
+  host::WindowHost* GetWindowHost() const override {
+    return window_host_.get();
+  }
 
   platform::gui::INativeWindow* GetNativeWindow();
 
@@ -30,8 +30,6 @@ class CRU_UI_API RootControl : public LayoutControl {
  private:
   std::unique_ptr<host::WindowHost> window_host_;
 
-  std::unique_ptr<render::StackLayoutRenderObject> render_object_;
-
   Control* attached_control_;
 
   EventRevokerListGuard