Added a few features to support load balancing sinks