1 files changed, 5 insertions, 3 deletions
diff --git a/test/test_windows.py b/test/test_windows.py
index cb7c86b7..af1b9d4a 100644
--- a/test/test_windows.py
+++ b/test/test_windows.py
@@ -9,11 +9,13 @@ config.config = ConfigShim()
 
 from poezio.windows import Input, HistoryInput, MessageInput
 
+class SubInput(Input):
+    def rewrite_text(self, *args, **kwargs):
+        return None
+
 @pytest.fixture
 def input():
-    input = Input()
-    input.rewrite_text = lambda: None
-    return input
+    return SubInput()
 
 class TestInput(object):