Function to fix states coded as countries