lib/prefixed_ids.rb in prefixed_ids-1.6.0 vs lib/prefixed_ids.rb in prefixed_ids-1.6.1

@@ -89,17 +89,19 @@
   module Finder
     extend ActiveSupport::Concern
 
     class_methods do
       def find(*ids)
-        prefix_ids = *ids.map do |id|
-          # Skip if model doesn't use prefixed ids
-          next id unless _prefix_id.present?
+        # Skip if model doesn't use prefixed ids
+        return super if _prefix_id.blank?
 
+        prefix_ids = ids.flatten.map do |id|
           prefix_id = _prefix_id.decode(id, fallback: _prefix_id_fallback)
           raise Error, "#{id} is not a valid prefix_id" if !_prefix_id_fallback && prefix_id.nil?
           prefix_id
         end
+        prefix_ids = [prefix_ids] if ids.first.is_a?(Array)
+
         super(*prefix_ids)
       end
 
       def relation
         super.tap { |r| r.extend ClassMethods }