lib/active_model/associations.rb in activemodel-associations-0.1.0 vs lib/active_model/associations.rb in activemodel-associations-0.1.1

@@ -16,11 +16,11 @@
 
     module ClassMethods
       # define association like ActiveRecord
       def belongs_to(name, scope = nil, options = {})
         reflection = ActiveRecord::Associations::Builder::BelongsTo.build(self, name, scope, options)
-        if ActiveRecord.version.to_s >= "4.1"
+        if ActiveRecord.version >= Gem::Version.new("4.1")
           ActiveRecord::Reflection.add_reflection self, name, reflection
         end
       end
 
       # define association like ActiveRecord
@@ -30,10 +30,10 @@
           options.merge!(scope)
           scope = nil
         end
 
         reflection = ActiveRecord::Associations::Builder::HasManyForActiveModel.build(self, name, scope, options, &extension)
-        if ActiveRecord.version.to_s >= "4.1"
+        if ActiveRecord.version >= Gem::Version.new("4.1")
           ActiveRecord::Reflection.add_reflection self, name, reflection
         end
 
         mixin = generated_association_methods
         mixin.class_eval <<-CODE, __FILE__, __LINE__ + 1