I need a small fix on a custom plugin I’ve built