Since Diane got arrested & reputation ruined, the police will have to believe High Voltage that she was the one who broke them out of jail which they planned to tell them. They can also tell them that she blackmailed them to steal & mutated them against their will.
Hopefully after returning to normal, High Voltage got off easy & went straight to community service (mentioned in S3) instead of jail right after Di got arrested with her crimes revealed. Everything they did wasn't their fault since they were controlled & blackmailed. They were victims.
Besides seeing Zeztz use Gravity's powers while in Catastrom, I've noticed that he used other Capsem's abilities without activating the Dual Nightmare Capsem & recently found out he used them while in Ordem Form too.
As much as I didn't want Di to live after all she did, I realized it makes sense for her to be arrested that way she can testify her crimes which could also help High Voltage & Knox since they were controlled & blackmailed into helping her & that she broke the dancing duo out of jail.