修改的判断不知道好不好使
Signed-off-by: manageryzy <manageryzy@live.cn>