diff --git a/src/agent/python/services.py b/src/agent/python/services.py
index 967eb3f..29a188a 100644
--- a/src/agent/python/services.py
+++ b/src/agent/python/services.py
@@ -155,7 +155,7 @@ async def _handle_client(self, reader: asyncio.StreamReader, writer: asyncio.Str
                     pair.comm_writer,
                     pair.data_reader,
                     pair.data_writer,
-                    get_data_source=lambda type: cast(IDataSource, self._extension_hive.get_extension_type(type)())
+                    get_data_source=lambda typeName: cast(IDataSource, self._extension_hive.get_extension_type(typeName)())
                 )
 
                 pair.task = self._create_task(pair.remote_communicator.run())