lib/mongoid/userstamp/railtie.rb in mongoid_userstamp-0.1.0 vs lib/mongoid/userstamp/railtie.rb in mongoid_userstamp-0.1.1

@@ -1,17 +1,17 @@
 # -*- encoding : utf-8 -*-
 module Mongoid
   module Userstamp
     class Railtie < Rails::Railtie
       ActiveSupport.on_load :action_controller do
-        unless Mongoid::Userstamp.configuration.user_model.respond_to? :current
-          Mongoid::Userstamp.configuration.user_model.send(
-            :include,
-            Mongoid::Userstamp::User
-          )
-        end
-
         before_filter do |c|
+          unless Mongoid::Userstamp.configuration.user_model.respond_to? :current
+            Mongoid::Userstamp.configuration.user_model.send(
+              :include,
+              Mongoid::Userstamp::User
+            )
+          end
+
           Mongoid::Userstamp.configuration.user_model.current = c.send(Mongoid::Userstamp.configuration.user_reader)
         end
       end
     end
   end