Move findModeRestoreSelection to FindMode::restoreSelection

1 files changed, 7 insertions, 7 deletions

diff --git a/content_scripts/vimium_frontend.coffee b/content_scripts/vimium_frontend.coffee
index b8bf9ef3..26eace9c 100644
--- a/content_scripts/vimium_frontend.coffee
+++ b/content_scripts/vimium_frontend.coffee
@@ -706,10 +706,16 @@ class FindMode extends Mode
     super()
     handleEscapeForFindMode() if event
 
+  restoreSelection: ->
+    range = @initialRange
+    selection = getSelection()
+    selection.removeAllRanges()
+    selection.addRange range
+
 window.performFindInPlace = ->
   # Restore the selection.  That way, we're always searching forward from the same place, so we find the right
   # match as the user adds matching characters, or removes previously-matched characters. See #1434.
-  findModeRestoreSelection()
+  findMode.restoreSelection()
   query = if findModeQuery.isRegex then getNextQueryFromRegexMatches(0) else findModeQuery.parsedQuery
   window.findModeQueryHasResults = executeFind(query, { caseSensitive: !findModeQuery.ignoreCase })
 
@@ -912,12 +918,6 @@ getCurrentRange = ->
     selection.collapseToStart() if selection.type == "Range"
     selection.getRangeAt 0
 
-findModeRestoreSelection = ->
-  range = findMode.initialRange
-  selection = getSelection()
-  selection.removeAllRanges()
-  selection.addRange range
-
 # Enters find mode.  Returns the new find-mode instance.
 window.enterFindMode = ->
   Marks.setPreviousPosition()