diff --git a/app/controllers/yiff_rest/api_v2_controller.rb b/app/controllers/yiff_rest/api_v2_controller.rb
index 6d55977..275f944 100644
--- a/app/controllers/yiff_rest/api_v2_controller.rb
+++ b/app/controllers/yiff_rest/api_v2_controller.rb
@@ -76,7 +76,7 @@ module YiffRest
       render(json: {
         success: true,
         data:    {
-          **APIImage.categories.find { |c| c[:db] == category },
+          **APIImage.categories.find { |c| c.db == category },
           count: count,
         },
       })
diff --git a/app/models/api_key.rb b/app/models/api_key.rb
index cebf99e..a8eb365 100644
--- a/app/models/api_key.rb
+++ b/app/models/api_key.rb
@@ -325,7 +325,7 @@ class APIKey < ApplicationRecord
   include WebhookMethods
 
   def self.stats(ip: nil, key: nil)
-    categories = APIImage.categories.pluck(:db)
+    categories = APIImage.categories.map(&:db)
 
     keys = [
       "yiffy2:stats:images:total",