Fix broken TestLayout