fixed build error cause the diagorm