end_of_association_chain support with policy_class scope

lib/simple_controller/base_controller.rb

@@ -170,8 +170,13 @@ class SimpleController::BaseController < ::InheritedResources::Base
   end
 
   def end_of_association_chain
+    policy_class ||= self.class.instance_variable_get(:@policy_class)
+    if policy_class.present? && scope_policy_class = "#{policy_class}::Scope".safe_constantize
+      after_association_chain(scope_policy_class.new(current_user, super).resolve)
+    else
       after_association_chain(super)
     end
+  end
 
   def collection
     get_collection_ivar || set_collection_ivar(