Fix mode key binding issue