1 files changed, 2 insertions, 10 deletions
diff --git a/src/ui/controls/Button.cpp b/src/ui/controls/Button.cpp
index e42bfae7..d7d157c5 100644
--- a/src/ui/controls/Button.cpp
+++ b/src/ui/controls/Button.cpp
@@ -10,11 +10,7 @@
 
 namespace cru::ui::controls {
 Button::Button() : click_detector_(this) {
-  render_object_ = std::make_unique<render::BorderRenderObject>();
-  render_object_->SetAttachedControl(this);
-  SetContainerRenderObject(render_object_.get());
-  render_object_->SetBorderEnabled(true);
-
+  GetContainerRenderObject()->SetBorderEnabled(true);
   auto default_button_style =
       ThemeManager::GetInstance()->GetResourceStyleRuleSet(u"button.style");
   GetStyleRuleSet()->SetParent(std::move(default_button_style));
@@ -22,11 +18,7 @@ Button::Button() : click_detector_(this) {
 
 Button::~Button() = default;
 
-render::RenderObject* Button::GetRenderObject() const {
-  return render_object_.get();
-}
-
 void Button::ApplyBorderStyle(const style::ApplyBorderStyleInfo& style) {
-  render_object_->ApplyBorderStyle(style);
+  GetContainerRenderObject()->ApplyBorderStyle(style);
 }
 }  // namespace cru::ui::controls