lib/devise/orm/mongo_mapper.rb in devise-0.9.2 vs lib/devise/orm/mongo_mapper.rb in devise-1.0.0

@@ -9,16 +9,16 @@
             super
           end
         end
       end
 
-      def self.included_modules_hook(klass, modules)
+      def self.included_modules_hook(klass)
         klass.send :extend,  self
         klass.send :include, InstanceMethods
         yield
 
-        modules.each do |mod|
+        klass.devise_modules.each do |mod|
           klass.send(mod) if klass.respond_to?(mod)
         end
       end
       
       def find(*args)
@@ -45,6 +45,6 @@
     end
   end
 end
 
 MongoMapper::Document::ClassMethods.send(:include, Devise::Models)
-MongoMapper::EmbeddedDocument::ClassMethods.send(:include, Devise::Models)
\ No newline at end of file
+MongoMapper::EmbeddedDocument::ClassMethods.send(:include, Devise::Models)