From 08c8937c962ab387f8fc373d73b4a39653d4f500 Mon Sep 17 00:00:00 2001
From: Jeff Casimir <jeff@casimircreative.com>
Date: Sat, 23 Jul 2011 09:34:06 -0700
Subject: [PATCH] Remove unnecessary class attribute

---
 lib/draper/base.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/draper/base.rb b/lib/draper/base.rb
index 95d24aa..bd4b836 100644
--- 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 @@ module Draper
 
     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