Fix layout break