Replace update_shared weight distribution