lib/enum_help/railtie.rb in enum_help-0.0.16 vs lib/enum_help/railtie.rb in enum_help-0.0.17

@@ -1,7 +1,9 @@
 module EnumHelp
   class Railtie < Rails::Railtie
     initializer "enum_help.i18n" do
-      ActiveRecord::Base.send :extend, EnumHelp::I18n
+      ActiveSupport.on_load(:active_record) do
+        extend EnumHelp::I18n
+      end
     end
   end
-end
\ No newline at end of file
+end