fix some tests

2 files changed, 2 insertions, 1 deletions

diff --git a/psutil/tests/test_connections.py b/psutil/tests/test_connections.py
index 93d8473a..c7fe1992 100755
--- a/psutil/tests/test_connections.py
+++ b/psutil/tests/test_connections.py
@@ -191,10 +191,10 @@ class TestUnconnectedSockets(Base, unittest.TestCase):
 
     def get_conn_from_sock(self, sock):
         cons = thisproc.connections(kind='all')
+        smap = dict([(c.fd, c) for c in cons])
         if NETBSD or FREEBSD:
             # NetBSD opens a UNIX socket to /var/log/run
             # so there may be more connections.
-            smap = dict([(c.fd, c) for c in cons])
             return smap[sock.fileno()]
         else:
             self.assertEqual(len(cons), 1)
diff --git a/psutil/tests/test_linux.py b/psutil/tests/test_linux.py
index f503b384..80bbd113 100755
--- a/psutil/tests/test_linux.py
+++ b/psutil/tests/test_linux.py
@@ -313,6 +313,7 @@ class TestSystemVirtualMemory(unittest.TestCase):
                 self.assertEqual(ret.available, 0)
                 self.assertEqual(ret.slab, 0)
 
+    @retry_on_failure()
     def test_avail_old_percent(self):
         # Make sure that our calculation of avail mem for old kernels
         # is off by max 10%.