diff --git a/client.py b/client.py
index 447e985..3e43fd0 100644
--- a/client.py
+++ b/client.py
@@ -18,7 +18,7 @@ class Client(PyOperator):
     """
     def __init__(self, url, model_name='pipeline'):
         super().__init__()
-        self._client = triton_client(url, model_name)
+        self._client = triton_client.Client(url, model_name)
 
     def __call__(self, *data):
         """