Fix on highlight on focus