Remove InvalidDimensionArgumentException