fix a bug
Signed-off-by: PacoIrene <zhenyangchu@gmail.com>