@Luka Gospodnetić Thank you for reaching out to us, for the same requirement we have sample custom policy which will help to achieve your desired result - https://github.com/azure-ad-b2c/samples/tree/master/policies/mfa-email-or-phone
Let me know if you have any further questions, feel free to post back.
Please remember to "Accept Answer" if answer helped, so that others in the community facing similar issues can easily find the solution.