e2e: Fix some logic in check_list_of_xpath

1 files changed, 3 insertions, 4 deletions

diff --git a/tests/end_to_end/__main__.py b/tests/end_to_end/__main__.py
index 4a42820..fedfa6a 100644
--- a/tests/end_to_end/__main__.py
+++ b/tests/end_to_end/__main__.py
@@ -151,14 +151,13 @@ def check_list_of_xpath(list_of_xpaths, xmpp, stanza):
     found = None
     for i, xpaths in enumerate(list_of_xpaths):
         if all_xpaths_match(stanza, xpaths):
-            found = i
+            found = True
+            list_of_xpaths.pop(i)
             break
 
-    if found is None:
+    if not found:
         raise StanzaError("Received stanza “%s” did not match any of the expected xpaths:\n%s" % (stanza, list_of_xpaths))
 
-    list_of_xpaths.pop(i)
-
     if list_of_xpaths:
         step = partial(expect_unordered_already_formatted, list_of_xpaths)
         xmpp.scenario.steps.insert(0, step)