fix tests for country