update code.

Signed-off-by: Jianmin Zhou <toandrew@infthink.com>