diff --git a/itkwidgets/_type_aliases.py b/itkwidgets/_type_aliases.py
index 95ddcf58..93cd2be6 100644
--- a/itkwidgets/_type_aliases.py
+++ b/itkwidgets/_type_aliases.py
@@ -20,7 +20,7 @@
 
 if HAVE_ITK:
     import itk
-    Image = Union[Image, itk.Image]
+    Image = Union[Image, itk.output]
 if HAVE_VTK:
     import vtk
     Image = Union[Image, vtk.vtkImageData]