Fix the code align