diff --git a/Labs/Made-up modules/lab_06.data_mem.sv b/Labs/Made-up modules/lab_06.data_mem.sv
index 02a9f2a..caf29a1 100644
--- a/Labs/Made-up modules/lab_06.data_mem.sv	
+++ b/Labs/Made-up modules/lab_06.data_mem.sv	
@@ -20,7 +20,7 @@ module data_mem(
   output logic        ready_o
 );
 assign ready_o = 1'b1;
-parameter DATA_MEM_SIZE_WORDS = 4096;
+import memory_pkg::DATA_MEM_SIZE_WORDS;
 logic [31:0] ram [DATA_MEM_SIZE_WORDS];
 
 always_ff @(posedge clk_i) begin