[MRG] properly handle tree weights for chains

previously only the weight of the first tree would be used.