tried to fix to make plugin work