Fix: Corrected focus issue on keydown code