diff --git a/spec/support/diagnostics_helper.rb b/spec/support/diagnostics_helper.rb
index 45a0f5146cc..bab725d8179 100644
--- a/spec/support/diagnostics_helper.rb
+++ b/spec/support/diagnostics_helper.rb
@@ -13,8 +13,9 @@ module MessageBus::DiagnosticsHelper
   def track_publish
     @tracking = tracking = []
     yield
-    @tracking = nil
     tracking
+  ensure
+    @tracking = nil
   end
 end