User Logs in and Variant doesn't change in checkout

We have a product experience where we have 2 product variants; one variant applies to retail users and the other variant is for wholesale pricing.

We are using Checkout Extensibility currently and offer the opportunity to customers to sign in and receive their wholesale price on our checkout page. The problem, however, is that the variant price doesn’t apply until the user navigates out of checkout, back to the cart and returns to the checkout.

Is there a way to force-apply the variant pricing during their checkout experience (after they log in) to not pull them out of the checkout experience?