Fix arrow keys in edit mode