Fix broken layout