fix the image button and icon