Wait for onDocumentFocus to happen before setting section range

Bug: 65928
Change-Id: If76408b6b89541f0ecb065d4860e0652b6bdebb5