Small update to plugin