[TASK] Layout fixes