调整代码

Signed-off-by: bee0060 <bee0060@126.com>