fix bug --> wrong tree