Fix around_save run_on_shard for custom connection with allow_octopus