Fix a little bug.

Signed-off-by: fatedier <512654112@qq.com>