Add image directory back in