Tree extension fix