Minor update to link function