Intelligent Account COmbination Defaulting is defaulting incorrect company code

Hello

We have multiple business units and we are testing the Intelligent Parables Account Combination Defaulting. However, the company code segment (which is the business unit number) is defaulting to an incorrect code even though the supplier site assignment is correct.

We have tried creating multiple invoices for this supplier in order to try and train the model but it is still defaulting to the incorrect company code.

Is there any other setup that I am missing?

Thanks,

Shriya