update code

Signed-off-by: leeyon <missile2006@126.com>