Fix plugin tests