diff --git a/python-api/gstswitch/test_connection.py b/python-api/gstswitch/test_connection.py
index 06f30eb..a818a81 100644
--- a/python-api/gstswitch/test_connection.py
+++ b/python-api/gstswitch/test_connection.py
@@ -134,8 +134,17 @@ def __init__(self, method):
         self.method = method
         self.return_result = self.funs[method]
 
-    def call_sync(self, a1, a2, a3, a4, a5, a6, a7, a8, a9):
-        if a3 == "info.duzy.gst.switch.SwitchControllerInterface":
+    def call_sync(
+            self,
+            bus_name,
+            object_path,
+            interface_name,
+            method_name,
+            parameters,
+            reply_type, flags,
+            timeout_msec,
+            cancellable):
+        if interface_name == "info.duzy.gst.switch.SwitchControllerInterface":
             return self.return_result
         else:
             raise GLib.GError('{0}: Test Failed'.format(self.method))
@@ -310,4 +319,4 @@ def test_mark_tracking():
         conn = Connection(default_interface=default_interface)
         conn.connection = MockConnection('mark_tracking')
         face = [(1, 1, 1, 1), (2, 2, 2, 2)]
-        assert conn.mark_tracking(face) is None
\ No newline at end of file
+        assert conn.mark_tracking(face) is None