fix code style
address review comment