wait, i actually get what you mean now. adding a PR in a sec.