Fix plugin implementation