Fix to get image version