fix key input function