Fix array size error