Fix bug in test plugin