Fix imports
It's a feature now, not a plugin