Small fix

Signed-off-by: PlutoniumHeart <zhitaoli.hust@gmail.com>