Update test for layout change