diff --git a/tests/test_discovery.py b/tests/test_discovery.py
index 1fa3f5f..2b2996f 100644
--- a/tests/test_discovery.py
+++ b/tests/test_discovery.py
@@ -42,7 +42,7 @@ async def test_reg_then_unreg(arb_addr):
 
         await trio.sleep(0.1)
         assert uid not in aportal.actor._registry
-        sockaddrs = actor._registry[uid]
+        sockaddrs = actor._registry.get(uid)
         assert not sockaddrs