Fix array index bug