Gemfile in reform-rails-0.2.1 vs Gemfile in reform-rails-0.2.2

@@ -18,15 +18,15 @@
 
 rails_version = ENV.fetch("RAILS_VERSION", "6.0.0")
 
 # bored of wrestling with rails...
 
-gem("mongoid", "< 7.0") unless rails_version.include?('6.0')
+gem("mongoid", "< 7.0") unless rails_version.to_i >= 6
 
 
 gem "activerecord", "~> #{rails_version}"
 gem "railties", "~> #{rails_version}"
-if rails_version.include?('6.0')
+if rails_version.to_i >= 6
   gem "sqlite3", "~> 1.4"
 else
   gem "sqlite3", "~> 1.3", "< 1.4"
 end
 puts "Rails version #{rails_version}"