diff --git a/src/gitsweep/tests/test_cli.py b/src/gitsweep/tests/test_cli.py
index 9ddd6ce..cf0f2f3 100644
--- a/src/gitsweep/tests/test_cli.py
+++ b/src/gitsweep/tests/test_cli.py
@@ -135,7 +135,7 @@ def test_will_cleanup(self):
             self.make_commit()
             self.command('git merge branch{0}'.format(i))
 
-        with patch('gitsweep.cli.raw_input', create=True) as ri:
+        with patch('gitsweep.cli.input', create=True) as ri:
             ri.return_value = 'y'
             (retcode, stdout, stderr) = self.gscommand('git-sweep cleanup')
 
@@ -173,7 +173,7 @@ def test_will_abort_cleanup(self):
             self.make_commit()
             self.command('git merge branch{0}'.format(i))
 
-        with patch('gitsweep.cli.raw_input', create=True) as ri:
+        with patch('gitsweep.cli.input', create=True) as ri:
             ri.return_value = 'n'
             (retcode, stdout, stderr) = self.gscommand('git-sweep cleanup')