Fix tree_image directory issue