Improved extracting flavour info from MC data