fix a bug

Signed-off-by: m208 <1010336298@qq.com>