改BUG

Signed-off-by: Wanghaoran <whrlmc@139.com>