diff --git a/test/test_ui.py b/test/test_ui.py
index 609c21c00..58eecee46 100644
--- a/test/test_ui.py
+++ b/test/test_ui.py
@@ -158,10 +158,13 @@ class ModifyTest(unittest.TestCase, TestHelper):
     def tearDown(self):
         self.teardown_beets()
 
-    def modify(self, *args):
-        with control_stdin('y'):
+    def modify_inp(self, inp, *args):
+        with control_stdin(inp):
             ui._raw_main(['modify'] + list(args), self.lib)
 
+    def modify(self, *args):
+        self.modify_inp('y', *args)
+
     # Item tests
 
     def test_modify_item(self):