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

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