Fix keyboard control