Hopefully fixed the align function