Flatten layer instead of merging them