diff --git a/lib/active_admin/sortable_tree/controller_actions.rb b/lib/active_admin/sortable_tree/controller_actions.rb
index e08df36..400f9e3 100644
--- a/lib/active_admin/sortable_tree/controller_actions.rb
+++ b/lib/active_admin/sortable_tree/controller_actions.rb
@@ -24,10 +24,12 @@ def sortable(options = {})
       collection_action :sort, :method => :post do
         resource_name = ActiveAdmin::SortableTree::Compatibility.normalized_resource_name(active_admin_config.resource_name)
 
+        resources = resource_class.where(id: params[resource_name].keys).map { |x| [x.id, x] }
+        resources = Hash[resources]
+
         records = []
         params[resource_name].each_pair do |resource, parent_resource|
-          parent_resource = resource_class.find(parent_resource) rescue nil
-          records << [resource_class.find(resource), parent_resource]
+          records << [resources[resource.to_i], resources[parent_resource.to_i]]
         end
 
         errors = []