lib/haml-rails.rb in haml-rails-0.3.3 vs lib/haml-rails.rb in haml-rails-0.3.4

@@ -2,10 +2,14 @@
 require 'rails'
 
 module Haml
   module Rails
     class Railtie < ::Rails::Railtie
-      config.generators.template_engine :haml
+      if ::Rails.version.to_f >= 3.1
+        config.app_generators.template_engine :haml
+      else
+        config.generators.template_engine :haml
+      end
 
       config.before_initialize do
         Haml.init_rails(binding)
         Haml::Template.options[:format] = :html5
       end