Fixes small bug in the layout