lib/partisan/railtie.rb in partisan-0.1.1 vs lib/partisan/railtie.rb in partisan-0.2

@@ -2,17 +2,9 @@
 require 'rails'
 
 module Partisan
   class Railtie < Rails::Railtie
     initializer "follows.active_record" do |app|
-      ActiveSupport.on_load :active_record do
-        def self.acts_as_follower
-          self.send :include, Partisan::Follower
-        end
-
-        def self.acts_as_followable
-          self.send :include, Partisan::Followable
-        end
-      end
+      ActiveSupport.on_load :active_record, {}, &Partisan.inject_into_active_record
     end
   end
 end