Removed minor bug in country allocation