minor cleanup.

thehappycoder · May 4, 2014 · 304cfad · 304cfad
1 parent 6f7c6bd
commit 304cfad
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lib/reform/contract.rb b/lib/reform/contract.rb
@@ -10,7 +10,7 @@ class Contract # DISCUSS: make class?
 
     extend Uber::InheritableAttr
     inheritable_attr :representer_class
-    self.representer_class = Reform::Representer.for(:form_class => Reform::Contract)
+    self.representer_class = Reform::Representer.for(:form_class => self)
 
     inheritable_attr :features
     self.features = []

diff --git a/lib/reform/form.rb b/lib/reform/form.rb
@@ -5,11 +5,11 @@
 
 module Reform
   class Form < Contract
-    self.representer_class = Reform::Representer.for(:form_class => Reform::Form)
+    self.representer_class = Reform::Representer.for(:form_class => self)
 
     def aliased_model
       # TODO: cache the Expose.from class!
-      Reform::Expose.from(self.class.representer_class).new(:model => model)
+      Reform::Expose.from(mapper).new(:model => model)
     end
 
     require "reform/form/virtual_attributes"