Fix layout update