Small fix that fixes new array dimension