Code alignment fix