Update code.

Signed-off-by: JiYou <jiyou09@gmail.com>