[editor] Fix selection logic to only use one method

..instead of multiple methods doing almost the same stuff.