lib/slim-rails.rb in slim-rails-0.2.1 vs lib/slim-rails.rb in slim-rails-1.0.0

@@ -2,13 +2,13 @@
 require 'slim'
 
 module Slim
   module Rails
     class Railtie < ::Rails::Railtie
-      if ::Rails.version.to_f >= 3.1
-        config.app_generators.template_engine :slim
-      else
+      unless ::Rails.version =~ /$3.1/
         config.generators.template_engine :slim
+      else
+        config.app_generators.template_engine :slim
       end
     end
   end
 end