Fixed an array dimension issue