Bug fix for the image button