diff --git a/lib/is_it_up/railtie.rb b/lib/is_it_up/railtie.rb
index 77cbf92..5d60d08 100644
--- a/lib/is_it_up/railtie.rb
+++ b/lib/is_it_up/railtie.rb
@@ -1,7 +1,7 @@
 module IsItUp
   class Railtie < Rails::Railtie
-    initializer "is_it_up.configure_rails_initialization" do
-      Rails.application.middleware.use IsItUp::Middleware
+    initializer "is_it_up.configure_rails_initialization" do |app|
+      app.middleware.use IsItUp::Middleware
     end
   end
 end