Mypy type correctness (#3199)

1 files changed, 4 insertions, 2 deletions

diff --git a/lib/elements/image.py b/lib/elements/image.py
index 695515dc..ad012975 100644
--- a/lib/elements/image.py
+++ b/lib/elements/image.py
@@ -23,14 +23,16 @@ class ImageObject(EmbroideryElement):
     name = "Image"
 
     def center(self):
-        transform = get_node_transform(self.node.getparent())
+        parent = self.node.getparent()
+        assert parent is not None, "This should be part of a tree and therefore have a parent"
+        transform = get_node_transform(parent)
         center = self.node.bounding_box(transform).center
         return center
 
     def validation_warnings(self):
         yield ImageTypeWarning(self.center())
 
-    def to_stitch_groups(self, last_stitch_group):
+    def to_stitch_groups(self, last_stitch_group, next_element):
         return []
 
     def first_stitch(self):