修改 key 类型判断