diff --git a/display.py b/display.py
index f43fe63..c102eb8 100644
--- a/display.py
+++ b/display.py
@@ -148,7 +148,7 @@ def _on_mouse_move(self, obj, event):
 
     def add(self, name, obj, color=None, opacity=None):
         mapper = vtkPolyDataMapper()
-        mapper.SetInput(obj.GetOutput())
+        mapper.SetInputConnection(obj.GetOutputPort())
         actor = vtkActor()
         actor.SetMapper(mapper)
         if color is not None:
@@ -196,7 +196,7 @@ def __init__(self):
 
     def __call__(self, source):
         filter = vtkTransformPolyDataFilter()
-        filter.SetInput(source.GetOutput())
+        filter.SetInputConnection(source.GetOutputPort())
         filter.SetTransform(self.transform)
         return filter
 
@@ -221,7 +221,7 @@ def __getitem__(self, index):
 
 def filter_from_transform(transform, source):
     filter = vtkTransformPolyDataFilter()
-    filter.SetInput(source.GetOutput())
+    filter.SetInputConnection(source.GetOutputPort())
     filter.SetTransform(transform)
     return filter