diff --git a/test/ganeti.client.gnt_instance_unittest.py b/test/ganeti.client.gnt_instance_unittest.py
index 51410958a043f23703388e06fb92f5b08db0f964..6326963a7ad2fe99949841c70cfd5e9391582235 100755
--- a/test/ganeti.client.gnt_instance_unittest.py
+++ b/test/ganeti.client.gnt_instance_unittest.py
@@ -38,8 +38,8 @@ class TestConsole(unittest.TestCase):
     self._cmds = []
     self._next_cmd_exitcode = 0
 
-  def _Test(self, *args):
-    return gnt_instance._DoConsole(*args,
+  def _Test(self, console, show_command, cluster_name):
+    return gnt_instance._DoConsole(console, show_command, cluster_name,
                                    feedback_fn=self._Feedback,
                                    _runcmd_fn=self._FakeRunCmd)