changed plugin to use new context arg