Thank you @happy-kitchen-89482!! I just updated it with the changes. I wasn't sure what you meant exactly by "a function that wraps it" in https://github.com/pantsbuild/pants/pull/4593/files/2d608e2fde346485b9410af14e3855357812e0c2#diff-8603ca893539bade8d782040a56d85b8 because I wasn't sure what "it" was referring to, so if you don't like it, let me know!