Fix bug in add image feature