Fixing alignment of diagrams