A method for imaging includes receiving a first three-dimensional (3D) image of a vicinity of an organ within a body of a subject and creating a geometrical model of a tubular structure in the first 3D image in the vicinity of the organ. An invasive probe is inserted into the organ, and a second 3D image containing the organ is captured using the invasive probe. One or more points are located on a surface of the tubular structure using the invasive probe, and the second 3D image is registered with the first 3D image by matching the one or more points to the geometrical model.