lib/switchman/standard_error.rb in switchman-2.0.6 vs lib/switchman/standard_error.rb in switchman-2.0.7

@@ -6,9 +6,9 @@
       @active_shards = Shard.send(:active_shards).dup
       super
     end
 
     def current_shard(category = :primary)
-      @active_shards[category] || Shard.default
+      @active_shards&.[](category) || Shard.default
     end
   end
 end