Minor code alignment fix