Updated test to use LayoutLR where possible.