Fix a little bug

Signed-off-by: zdd <zddhub@gmail.com>