Fix invalid index error