Old balancing code put back  for bug diagnosis