diff --git a/socrates/dma350/config/address_map_m1_example2.sv b/socrates/dma350/config/address_map_m1_cocotb.sv
similarity index 90%
rename from socrates/dma350/config/address_map_m1_example2.sv
rename to socrates/dma350/config/address_map_m1_cocotb.sv
index 6bb87fa3761a0990d21adeb6ccabb59e766b92b7..152a291d9b2837d0fcef45b7fe1a113cc59a7b98 100755
--- a/socrates/dma350/config/address_map_m1_example2.sv
+++ b/socrates/dma350/config/address_map_m1_cocotb.sv
@@ -22,10 +22,8 @@ function automatic bit address_map_m1 (
 );
 bit res;
 begin
-  res = '1;
-  if (axaddr[31:20] == 12'h000 || axaddr[31:20] == 12'h200) begin
-    res = '0;
-  end
+  res = '0;
+  if (axaddr[27] == '1) res = '1;
   return res;
 end
 endfunction
diff --git a/socrates/dma350/config/cfg_dma_axi.yaml b/socrates/dma350/config/cfg_dma_axi.yaml
index 5857ba6fabbae35ef8db1a639d289b2b0317af4a..100945b51dae48971c1a11010c99c73f7c9597d2 100755
--- a/socrates/dma350/config/cfg_dma_axi.yaml
+++ b/socrates/dma350/config/cfg_dma_axi.yaml
@@ -162,4 +162,4 @@ SECEXT_PRESENT: 0
 #
 #     Valid values:
 #         relative path to logical
-AXI5_M1_ADDR_MAP: models/modules/generic/address_map_m1_nanosoc.sv
+AXI5_M1_ADDR_MAP: models/modules/generic/address_map_m1_cocotb.sv