diff --git a/picosoc/icebreaker.v b/picosoc/icebreaker.v
index 00da2d7..342f346 100644
--- a/picosoc/icebreaker.v
+++ b/picosoc/icebreaker.v
@@ -45,6 +45,8 @@ module icebreaker (
 	inout  flash_io2,
 	inout  flash_io3
 );
+	parameter integer MEM_WORDS = 32768;
+
 	reg [5:0] reset_cnt = 0;
 	wire resetn = &reset_cnt;
 
@@ -107,7 +109,7 @@ module icebreaker (
 	picosoc #(
 		.BARREL_SHIFTER(0),
 		.ENABLE_MULDIV(0),
-		.MEM_WORDS(32768)
+		.MEM_WORDS(MEM_WORDS)
 	) soc (
 		.clk          (clk         ),
 		.resetn       (resetn      ),
diff --git a/picosoc/icebreaker_tb.v b/picosoc/icebreaker_tb.v
index 2126ca2..209d165 100644
--- a/picosoc/icebreaker_tb.v
+++ b/picosoc/icebreaker_tb.v
@@ -62,7 +62,12 @@ module testbench;
 		#1 $display("%b", leds);
 	end
 
-	icebreaker uut (
+	icebreaker #(
+		// We limit the amount of memory in simulation
+		// in order to avoid reduce simulation time
+		// required for intialization of RAM
+		.MEM_WORDS(256)
+	) uut (
 		.clk      (clk      ),
 		.led1     (led1     ),
 		.led2     (led2     ),