diff --git a/spec/components/discourse_redis_spec.rb b/spec/components/discourse_redis_spec.rb
index 6f0f1619b..fd0435828 100644
--- a/spec/components/discourse_redis_spec.rb
+++ b/spec/components/discourse_redis_spec.rb
@@ -83,8 +83,10 @@ describe DiscourseRedis do
         begin
           fallback_handler.master = false
           Redis::Client.any_instance.expects(:call).with([:info]).returns(DiscourseRedis::FallbackHandler::MASTER_LINK_STATUS)
-          Redis::Client.any_instance.expects(:call).with([:client, [:kill, 'type', 'normal']])
-          Redis::Client.any_instance.expects(:call).with([:client, [:kill, 'type', 'pubsub']])
+
+          %w{normal pubsub}.each do |connection_type|
+            Redis::Client.any_instance.expects(:call).with([:client, [:kill, 'type', connection_type]])
+          end
 
           fallback_handler.initiate_fallback_to_master