diff --git a/lib/zip_kit/railtie.rb b/lib/zip_kit/railtie.rb
index bc82a38..ca685f9 100644
--- a/lib/zip_kit/railtie.rb
+++ b/lib/zip_kit/railtie.rb
@@ -2,6 +2,8 @@
 
 class ZipKit::Railtie < ::Rails::Railtie
   initializer "zip_kit.install_extensions" do |app|
-    ActionController::Base.include(ZipKit::RailsStreaming)
+    ActiveSupport.on_load(:action_controller) do
+      include(ZipKit::RailsStreaming)
+    end
   end
 end