--- a/test/test_nicclient.py	Thu Jun 02 13:59:46 2016 -0700
+++ b/test/test_nicclient.py	Thu Jun 02 14:05:21 2016 -0700
@@ -18,7 +18,6 @@
     def test_choose_server(self):
         domain = 'рнидс.срб'
         chosen = self.client.choose_server(domain)
-        self.assertEqual(
-            chosen,
-            'срб'.decode('utf-8').encode('idna') + '.whois-servers.net'
-        )
+        suffix = domain.split('.')[-1].encode('idna').decode('utf-8')
+        correct = '{}.whois-servers.net'.format(suffix)
+        self.assertEqual(chosen, correct)