Fixed the plugin

Runs after persistence phase, can finally get scalameta trees correctly