optimize code style

Signed-off-by: Tiger Wang <wanghu-520@163.com>