Fix array logic error