Fix IndexError in array utils