Update to new plugin method