small tweaks

Signed-off-by: Wang Bin <wangbin.zibo@gmail.com>