update plugin to actually work properly