Update function definition of tps_segment_registration