Remove unnecessary class attribute

imesmes · Jul 23, 2011 · 08c8937 · 08c8937
1 parent 605332c
commit 08c8937
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/draper/base.rb b/lib/draper/base.rb
@@ -1,7 +1,7 @@
 module Draper
   class Base      
     require 'active_support/core_ext/class/attribute'
-    class_attribute :denied, :allowed, :source_class, :model_class
+    class_attribute :denied, :allowed, :model_class
     attr_accessor :model
 
     DEFAULT_DENIED = Object.new.methods
@@ -10,7 +10,7 @@ class Base
 
     def initialize(input)
       input.inspect
-      self.class.source_class = input.class
+      self.class.model_class = input.class if model_class.nil?
       @model = input      
       build_methods
     end