minor fix in code alignment