添加 type 判断