Updated method signature to match with the latest changes on the plugin