FIX: plugin wasn't working due to recent changes and ES6ification