Fix context in layout_start