diff --git a/DRAMSys/simulator/src/simulation/Arbiter.h b/DRAMSys/simulator/src/simulation/Arbiter.h
index db8c46af..76939a08 100644
--- a/DRAMSys/simulator/src/simulation/Arbiter.h
+++ b/DRAMSys/simulator/src/simulation/Arbiter.h
@@ -137,9 +137,9 @@ private:
 
     virtual unsigned int transport_dbg(int id, tlm::tlm_generic_payload &trans)
     {
-        //appendDramExtension(id, trans);
-        //trans.acquire();
-        return iSocket[id]->transport_dbg(trans);
+        DecodedAddress decodedAddress = xmlAddressDecoder::getInstance().decodeAddress(trans.get_address());
+        return iSocket[decodedAddress.channel]->transport_dbg(trans);
+
     }
 
     void peqCallback(tlm_generic_payload& payload, const tlm_phase& phase)