diff --git a/lib/app_profiler/railtie.rb b/lib/app_profiler/railtie.rb
index 7e616b9..7cfb6e8 100644
--- a/lib/app_profiler/railtie.rb
+++ b/lib/app_profiler/railtie.rb
@@ -68,7 +68,7 @@ class Railtie < Rails::Railtie
     private
 
     def default_middleware_action
-      if Rails.env.development? || Rails.env.test?
+      if Rails.env.development?
         Middleware::ViewAction
       else
         Middleware::UploadAction