Fix image size bug