2 files changed, 8 insertions, 5 deletions
diff --git a/activerecord/lib/active_record/identity_map.rb b/activerecord/lib/active_record/identity_map.rb
index fe448abfdf..30a2a1eb0f 100644
--- a/activerecord/lib/active_record/identity_map.rb
+++ b/activerecord/lib/active_record/identity_map.rb
@@ -98,8 +98,9 @@ module ActiveRecord
       end
 
       def call(env)
-        ActiveRecord::IdentityMap.enabled? ?
-          ActiveRecord::IdentityMap.use { @app.call(env) } : @app.call(env)
+        ActiveRecord::IdentityMap.use do
+          @app.call(env)
+        end
       end
     end
   end
diff --git a/activerecord/lib/active_record/railtie.rb b/activerecord/lib/active_record/railtie.rb
index 5d0424a543..6afa2d4cf3 100644
--- a/activerecord/lib/active_record/railtie.rb
+++ b/activerecord/lib/active_record/railtie.rb
@@ -22,9 +22,6 @@ module ActiveRecord
     config.app_middleware.insert_after "::ActionDispatch::Callbacks",
       "ActiveRecord::ConnectionAdapters::ConnectionManagement"
 
-    config.app_middleware.insert_after "::ActionDispatch::Callbacks",
-      "ActiveRecord::IdentityMap::Middleware"
-
     rake_tasks do
       load "active_record/railties/databases.rake"
     end
@@ -81,6 +78,11 @@ module ActiveRecord
       end
     end
 
+    initializer "active_record.identity_map" do |app|
+      config.app_middleware.insert_after "::ActionDispatch::Callbacks",
+        "ActiveRecord::IdentityMap::Middleware" if ActiveRecord::IdentityMap.enabled?
+    end
+
     config.after_initialize do
       ActiveSupport.on_load(:active_record) do
         instantiate_observers