1 files changed, 7 insertions, 5 deletions
diff --git a/src/theme_builder/components/MainWindow.h b/src/theme_builder/components/MainWindow.h
index 101b678e..c6733111 100644
--- a/src/theme_builder/components/MainWindow.h
+++ b/src/theme_builder/components/MainWindow.h
@@ -3,6 +3,7 @@
 #include "cru/ui/controls/Button.h"
 #include "cru/ui/controls/FlexLayout.h"
 #include "cru/ui/controls/StackLayout.h"
+#include "cru/ui/controls/TextBlock.h"
 #include "cru/ui/controls/Window.h"
 
 namespace cru::theme_builder {
@@ -15,14 +16,15 @@ class MainWindow : public ui::components::Component {
 
   ~MainWindow() override;
 
-  ui::controls::Control* GetRootControl() override { return window_; }
+  ui::controls::Control* GetRootControl() override { return &window_; }
 
   void Show();
 
  private:
-  ui::controls::Window* window_;
-  ui::controls::FlexLayout* main_layout_;
-  ui::controls::StackLayout* preview_layout_;
-  ui::controls::Button* preview_button_;
+  ui::controls::Window window_;
+  ui::controls::FlexLayout main_layout_;
+  ui::controls::StackLayout preview_layout_;
+  ui::controls::Button preview_button_;
+  ui::controls::TextBlock preview_button_text_;
 };
 }  // namespace cru::theme_builder