A method includes, in a processor, receiving position signals that are indicative of positions of (i) multiple electrodes disposed on an inflatable balloon fitted at a distal end of a catheter, and (ii) first and second electrodes fitted on a shaft of the catheter, on either side of the balloon. The positions of the multiple electrodes disposed on the balloon are calculated based on the received position signals and based on a known distance between the first and second electrodes.