Yup, I've updated that method and re-pushed. Makes sense. Unfortunately I can't validate it locally but hopefully it will work in sandbox.