diff --git a/lib/CPUs/JCpu/src/itest_irom.vhdl b/lib/CPUs/JCpu/src/itest_irom.vhdl
index 94609a3..0ed28f2 100644
--- a/lib/CPUs/JCpu/src/itest_irom.vhdl
+++ b/lib/CPUs/JCpu/src/itest_irom.vhdl
@@ -26,25 +26,22 @@ use IEEE.STD_LOGIC_1164.ALL;
 USE ieee.numeric_std.ALL;
 
 library work;
-use work.types_pkg.all;
+use work.cpu_pkg.all;
 
 -- JASM_ROM_INSERT_HERE
 ENTITY irom IS
-	Generic 
+	Port
 	(
-		depth 	: integer := 4
-	);
-	Port (
 	clk			: in  STD_LOGIC;
 	ce			: in  STD_LOGIC;
-	addr		: in  addr_t;
-	dout		: out word_t
+	addr		: in  inst_addr_t;
+	dout		: out inst_t
 	);
 END irom;
 
 ARCHITECTURE itest OF irom IS 
 
-	type imem_rom_t is array (0 to depth-1) of word_t;
+	type imem_rom_t is array (0 to 1023) of inst_t;
 
 	-- Assembled from itest.jsm
 	constant imem_rom : imem_rom_t :=