Fix layout problem