diff --git a/lib/facebook_dialog/railtie.rb b/lib/facebook_dialog/railtie.rb
index d098df8..e701a12 100644
--- a/lib/facebook_dialog/railtie.rb
+++ b/lib/facebook_dialog/railtie.rb
@@ -2,7 +2,7 @@
 
 module FacebookDialog
   class Railtie < Rails::Railtie
-    ActiveSupport.on_load(:action_controller) do
+    ActiveSupport.on_load(:action_controller_base) do
       helper FacebookDialog::ScriptHelper
       if FacebookDialog.omniauth_enabled?
         require "facebook_dialog/omni_auth_helper"