Fixing error with countries