Fix alignment bug