modify code

Signed-off-by: 吴炅明 <wu922008@126.com>