Fix layout code