Properly recalculate tax after changing country or state in checkout