fix weird highlight issue