fix index error bug