fix key bug