Fix tree test

Weighting is only 1 in the simplest case, not the usual case.