lib/brillo/scrubber.rb in brillo-1.3.1 vs lib/brillo/scrubber.rb in brillo-2.0.0.rc1

@@ -19,11 +19,11 @@
         end.each(&:capitalize!).join(' ')
       },
     }
 
     TACTICS = {
-      latest: -> (klass) { klass.order('id desc').limit(LATEST_LIMIT).pluck(:id) },
-      all:    -> (klass) { klass.pluck(:id) }
+      latest: -> (klass) { klass.order("#{klass.primary_key} desc").limit(LATEST_LIMIT).pluck(klass.primary_key) },
+      all:    -> (klass) { klass.pluck(klass.primary_key) }
     }
 
     attr_reader :config, :adapter, :transferrer
 
     def initialize(config)