Fix keyboard controls