Sneaking this fix in. When switching documents that are live, we should immediately restore the highlight given the current cursor position in the new doc.