lib/draper/finders.rb in draper-1.2.0 vs lib/draper/finders.rb in draper-1.2.1

@@ -3,29 +3,29 @@
   # do not have to extend this module directly; it is extended by
   # {Decorator.decorates_finders}.
   module Finders
 
     def find(id, options = {})
-      decorate(source_class.find(id), options)
+      decorate(object_class.find(id), options)
     end
 
     def all(options = {})
-      decorate_collection(source_class.all, options)
+      decorate_collection(object_class.all, options)
     end
 
     def first(options = {})
-      decorate(source_class.first, options)
+      decorate(object_class.first, options)
     end
 
     def last(options = {})
-      decorate(source_class.last, options)
+      decorate(object_class.last, options)
     end
 
     # Decorates dynamic finder methods (`find_all_by_` and friends).
     def method_missing(method, *args, &block)
       return super unless method =~ /^find_(all_|last_|or_(initialize_|create_))?by_/
 
-      result = source_class.send(method, *args, &block)
+      result = object_class.send(method, *args, &block)
       options = args.extract_options!
 
       if method =~ /^find_all/
         decorate_collection(result, options)
       else