Reset focus on keydown, not click

Fixes #1