yeah, good point. I'll update the PR shortly