Fix the style issue of code