You're so close! Instead of patching the issue maybe let's solve the root problem of spiky power distribution among humans. We don't need to make sure cops have immunity to prosecute powerful people. We need to not have powerful people.

(though realistically speaking yes there's probably some level of procedural immunity that probably makes sense, similarly with business bankruptcies not ruining the people who start the business)