Fixed the plugin