Fixed error that caused the plugin to fail