From fc9f12d6d7ae6bd4cb25d34c51fc98fea67222c4 Mon Sep 17 00:00:00 2001
From: Andrew Morgan <andrew@amorgan.xyz>
Date: Fri, 30 Aug 2019 13:58:01 +0100
Subject: [PATCH] Fix broken parentheses removal

---
 tests/test_state.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_state.py b/tests/test_state.py
index c404c0d9ca..610ec9fb46 100644
--- a/tests/test_state.py
+++ b/tests/test_state.py
@@ -152,7 +152,7 @@ class Graph(object):
         return iter(self._events)
 
     def get_leaves(self):
-        return self._events[i] for i in self._leaves
+        return (self._events[i] for i in self._leaves)
 
 
 class StateTestCase(unittest.TestCase):