Use correct visual mode in function text object