diff --git a/exploit.py b/exploit.py
index 39b708a..7b5cbfe 100644
--- a/exploit.py
+++ b/exploit.py
@@ -28,6 +28,8 @@
 from pyrdp.parser import NegotiationRequestParser, NegotiationResponseParser, ClientConnectionParser, GCCParser, \
   ServerConnectionParser
 from pyrdp.mcs import MCSClientChannel
+from pyrdp.logging import LOGGER_NAMES, SessionLogger
+import logging
 
 # Hard-coded constant
 PAYLOAD_HEAD_ADDR = 0xfffffa8008711010 + 0x38
@@ -53,9 +55,10 @@ def __init__(self, _reactor: Reactor, host: str, port: int, ip: str, bport: int)
       onDemandActive = self.onDemandActive, 
       onData = self.onSlowPathPDUReceived
     )
-
+    logger = logging.getLogger(LOGGER_NAMES.MITM_CONNECTIONS)
+    log = SessionLogger(logger, "replay")
     config = MITMConfig()
-    self.state = RDPMITMState(config)
+    self.state = RDPMITMState(config, log.sessionID)
     self.extra_channels = []
     self.name_to_id = {}
     self.virt_layer = {}