1 files changed, 7 insertions, 4 deletions
diff --git a/tests/test_elements_utils.py b/tests/test_elements_utils.py
index bdb78003..4653ac69 100644
--- a/tests/test_elements_utils.py
+++ b/tests/test_elements_utils.py
@@ -1,8 +1,11 @@
-from lib.elements import utils, FillStitch
-from inkex import Rectangle, Group, Style
+from inkex import Group, Rectangle, Style
 from inkex.tester import TestCase
 from inkex.tester.svg import svg
 
+from lib.elements import FillStitch, utils
+
+from .utils import element_count
+
 
 class ElementsUtilsTest(TestCase):
     # These tests test two functions at once, but they're sort of complimentary.
@@ -28,7 +31,7 @@ class ElementsUtilsTest(TestCase):
         }))
 
         elements = utils.nodes_to_elements(utils.iterate_nodes(g))
-        self.assertEqual(len(elements), 2)
+        self.assertEqual(len(elements), element_count())
         self.assertEqual(type(elements[0]), FillStitch)
         self.assertEqual(elements[0].node, rect)
 
@@ -41,7 +44,7 @@ class ElementsUtilsTest(TestCase):
         }))
 
         elements = utils.nodes_to_elements(utils.iterate_nodes(rect))
-        self.assertEqual(len(elements), 2)
+        self.assertEqual(len(elements), element_count())
         self.assertEqual(type(elements[0]), FillStitch)
         self.assertEqual(elements[0].node, rect)