Fix code alignment