Fix plugin bug