Fix ModelTree bugs