Fix bug in country model