From 72c2b472225b5a0d7c952c84ac0bee871d97baf3 Mon Sep 17 00:00:00 2001 From: jMyles Date: Fri, 16 Nov 2018 14:16:55 -0800 Subject: [PATCH] Let's allow FleetStateTracker to be checked for addresses, not just VerifiableNodes. --- nucypher/network/nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nucypher/network/nodes.py b/nucypher/network/nodes.py index f08fe5107..fa923bc32 100644 --- a/nucypher/network/nodes.py +++ b/nucypher/network/nodes.py @@ -111,7 +111,7 @@ class FleetStateTracker: return bool(self._nodes) def __contains__(self, item): - return item in self._nodes.values() + return item in self._nodes.keys() or item in self._nodes.values() def __iter__(self): yield from self._nodes.values()