diff --git a/spec/support/diagnostics_helper.rb b/spec/support/diagnostics_helper.rb
index b4c4e7e7382d9..4646857435670 100644
--- a/spec/support/diagnostics_helper.rb
+++ b/spec/support/diagnostics_helper.rb
@@ -7,6 +7,7 @@ def publish(channel, data, opts = nil)
       m = MessageBus::Message.new(-1, id, channel, data)
       m.user_ids = opts[:user_ids] if opts
       m.group_ids = opts[:group_ids] if opts
+      m.client_ids = opts[:client_ids] if opts
       @tracking << m
     end
     id