diff --git a/activerecord/lib/active_record/middleware/database_selector/resolver.rb b/activerecord/lib/active_record/middleware/database_selector/resolver.rb
index 930ddf900e6831b1ab976e179aed0e0d98b10e18..27050c9f2d54089109af865452665f60f316165f 100644
--- a/activerecord/lib/active_record/middleware/database_selector/resolver.rb
+++ b/activerecord/lib/active_record/middleware/database_selector/resolver.rb
@@ -64,12 +64,9 @@ def read_from_replica(&blk)
           def write_to_primary(&blk)
             ActiveRecord::Base.connected_to(role: :writing) do
               instrumenter.instrument("database_selector.active_record.wrote_to_primary") do
-                begin
-                  ret = yield
-                ensure
-                  resolver.update_last_write_timestamp
-                end
-                ret
+                yield
+              ensure
+                resolver.update_last_write_timestamp
               end
             end
           end