diff --git a/dsmr_parser/clients/protocol.py b/dsmr_parser/clients/protocol.py
index 4e6d85d..e996423 100644
--- a/dsmr_parser/clients/protocol.py
+++ b/dsmr_parser/clients/protocol.py
@@ -21,7 +21,7 @@ def create_dsmr_protocol(dsmr_version, telegram_callback, loop=None, **kwargs):
     return protocol
 
 
-def _create_dsmr_protocol(dsmr_version, telegram_callback, protocol loop=None, **kwargs):
+def _create_dsmr_protocol(dsmr_version, telegram_callback, protocol, loop=None, **kwargs):
     """Creates a DSMR asyncio protocol."""
 
     if dsmr_version == '2.2':
diff --git a/dsmr_parser/clients/rfxtrx_protocol.py b/dsmr_parser/clients/rfxtrx_protocol.py
index b8a347d..281f4c2 100644
--- a/dsmr_parser/clients/rfxtrx_protocol.py
+++ b/dsmr_parser/clients/rfxtrx_protocol.py
@@ -1,15 +1,13 @@
 """Asyncio protocol implementation for handling telegrams over a RFXtrx connection ."""
 
-from functools import partial
 import asyncio
 
 from serial_asyncio import create_serial_connection
-
 from .protocol import DSMRProtocol, _create_dsmr_protocol
 
 
 def create_rfxtrx_dsmr_protocol(dsmr_version, telegram_callback, loop=None, **kwargs):
-    """Creates a DSMR asyncio protocol."""
+    """Creates a RFXtrxDSMR asyncio protocol."""
     protocol = _create_dsmr_protocol(dsmr_version, telegram_callback,
                                      RFXtrxDSMRProtocol, loop, **kwargs)
     return protocol