fixed sth too.

Signed-off-by: wjk <478286460@qq.com>