Minor adjustment to plugin