Fix array bug I introduced