Update import statement for new plugin version