Minor update to plugin