Fix selection not being selected properly.