Move the imputer to the plugin