Forgot to add in the new plugin class