Fix for key input