A system includes an expandable distal-end assembly, a proximal position sensor, a distal position sensor, and a processor. The expandable distal-end assembly is coupled to a distal end of a shaft for insertion into a cavity of an organ of a patient. The proximal and distal position sensors are located at a proximal end and a distal end of the distal-end assembly, respectively. The processor is configured to estimate a position and a longitudinal direction of the proximal sensor, and a position of the distal sensor, all in a coordinate system used by the processor. The processor is further configured to project the estimated position of the distal sensor on an axis defined by the estimated longitudinal direction, and calculate an elongation of the distal-end assembly by calculating a distance between the estimated position of the proximal sensor and the projected position of the distal sensor.