Share tab payments allocated against correct budget categories

When someone pays you on the shared tab, it would be fantastic if Monzo allocated the amount based on how the shared transactions being paid were categorised.

Eg, Say I put a bunch of holiday flights on a shared tab, mixed in with some grocery shopping and bills. When my friend pays me back, Instead of the lump sum payment being put towards one category, it would be great if the payment was split out between holidays, groceries and bills based on the original amounts put into the share tab.