Fix bug when run the rspec

mashiro · May 8, 2012 · 1f35def · 1f35def
1 parent 4ef395d
commit 1f35def
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/sakuramochi/predicate_builder.rb b/lib/sakuramochi/predicate_builder.rb
@@ -7,9 +7,9 @@ module PredicateBuilder
 
     included do
       unless respond_to? :build
-        model_class = defined?(ActiveRecord::Model) ? ActiveRecord::Model : ActiveRecord::Base
-
         def self.build(attribute, value)
+          model_class = defined?(ActiveRecord::Model) ? ActiveRecord::Model : ActiveRecord::Base
+
           case value
           when ActiveRecord::Relation
             value = value.select(value.klass.arel_table[value.klass.primary_key]) if value.select_values.empty?