Minor fix in key_map