Fix tree extension