Fix jump to definition functionality