small refactor in the layout