Fix vertical alignment

But still has a bug in horizontal alignment