Updating methods signature to match plugin's api