diff --git a/cocotbext/axi/axi_master.py b/cocotbext/axi/axi_master.py index 9c63e94..9d144a1 100644 --- a/cocotbext/axi/axi_master.py +++ b/cocotbext/axi/axi_master.py @@ -128,7 +128,7 @@ class AxiMasterWrite(object): token = object() self.init_write(address, data, burst, size, lock, cache, prot, qos, region, user, token) await self.wait_for_token(token) - return self.get_write_resp(token) + return self.get_write_resp(token)[1:3] async def _process_write(self): while True: @@ -371,7 +371,7 @@ class AxiMasterRead(object): token = object() self.init_read(address, length, burst, size, lock, cache, prot, qos, region, user, token) await self.wait_for_token(token) - return self.get_read_data(token) + return self.get_read_data(token)[1:3] async def _process_read(self): while True: diff --git a/cocotbext/axi/axil_master.py b/cocotbext/axi/axil_master.py index 704363d..ac6642e 100644 --- a/cocotbext/axi/axil_master.py +++ b/cocotbext/axi/axil_master.py @@ -119,7 +119,7 @@ class AxiLiteMasterWrite(object): token = object() self.init_write(address, data, prot, token) await self.wait_for_token(token) - return self.get_write_resp(token) + return self.get_write_resp(token)[1:2] async def _process_write(self): while True: @@ -286,7 +286,7 @@ class AxiLiteMasterRead(object): token = object() self.init_read(address, length, prot, token) await self.wait_for_token(token) - return self.get_read_data(token) + return self.get_read_data(token)[1:2] async def _process_read(self): while True: