diff --git a/lib/generators/ruby_ui/install/install_generator.rb b/lib/generators/ruby_ui/install/install_generator.rb
index 23569f86..285855fa 100644
--- a/lib/generators/ruby_ui/install/install_generator.rb
+++ b/lib/generators/ruby_ui/install/install_generator.rb
@@ -42,7 +42,7 @@ def install_ruby_ui_initializer
 
       def add_ruby_ui_module_to_components_base
         say "Adding RubyUI Kit to Components::Base"
-        insert_into_file Rails.root.join("app/components/base.rb"), after: "include Components" do
+        insert_into_file Rails.root.join("app/components/base.rb"), after: "class Components::Base < Phlex::HTML" do
           "\n  include RubyUI"
         end
       end