修复上一次的逻辑问题

Signed-off-by: gaubee <gaubeebangeel@gmail.com>