Set proper regime when alternative code is given

CHANGELOG.md

@@ -15,6 +15,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/) and this p
 
 - `bill.Invoice` - remove empty taxes instances.
 - `tax.Identity` - support Calculate method to normalize IDs.
+- `tax.Regime` - properly set regime when alternative codes is given.
 
 ## [v0.202.0]
 

tax/regime.go

@@ -26,11 +26,12 @@ func (r Regime) GetRegime() l10n.TaxCountryCode {
 // that the regime is actually defined. Missing regimes will silently replace
 // the current regime with an empty value.
 func (r *Regime) SetRegime(country l10n.TaxCountryCode) {
-	if Regimes().For(country.Code()) == nil {
+	rd := Regimes().For(country.Code())
+	if rd == nil {
 		r.Country = ""
 		return
 	}
-	r.Country = country
+	r.Country = rd.Country
 }
 
 // RegimeDef provides the associated regime definition.

tax/regimes_test.go

@@ -19,3 +19,9 @@ func TestRegimesAltCountryCodes(t *testing.T) {
 	r := tax.RegimeDefFor("GR")
 	assert.Equal(t, "EL", r.Country.String())
 }
+
+func TestSetRegime(t *testing.T) {
+	r := new(tax.Regime)
+	r.SetRegime("GR")
+	assert.Equal(t, "EL", r.GetRegime().String())
+}