lib/prospecto/railtie.rb in prospecto-0.0.5 vs lib/prospecto/railtie.rb in prospecto-0.0.7

@@ -1,10 +1,12 @@
+require 'rails/railtie'
 require 'prospecto/controller'
 
 module Prospecto
   class Railtie < Rails::Railtie
     initializer "prospecto.initialize" do |app|
       ActionController::Base.extend(ProspectoController)
+      ActionController::Base.class_eval { enable_prospecto }
     end
 
     initializer 'prospecto.autoload', :before => :set_autoload_paths do |app|
       Dir["#{app.root}/app/views/**/"].each do |view_folder|
         app.config.autoload_paths << view_folder