lib/attribute_ext/railtie.rb in attribute_ext-1.1.0 vs lib/attribute_ext/railtie.rb in attribute_ext-1.2.4

@@ -1,11 +1,10 @@
 
 module AttributeExt
-  class Railtie < Rails::Railtie
+  class Railtie < Rails::Railtie # :nodoc:
     initializer 'attribute_ext' do |app|
       ActiveSupport.on_load :active_record do
-        ActiveRecord::Base.send :include, AttributeExt::HiddenAttributes
-        ActiveRecord::Base.send :include, AttributeExt::SafeAttributes
+        AttributeExt.setup
       end
     end
   end
 end
\ No newline at end of file