Correct bug in array size