Fix layout regression