fix code style
fix test problem