diff --git a/checks/rvfi_insn_check.py b/checks/rvfi_insn_check.py
index efc965a..5a9b4cf 100644
--- a/checks/rvfi_insn_check.py
+++ b/checks/rvfi_insn_check.py
@@ -1,7 +1,10 @@
 from nmigen import *
 from nmigen.hdl.ast import *
-# TODO: Find a way to make this import work
-# from ..insns.insn_add import *
+
+# Ugly hack for now to get the insn_add import working
+import sys
+sys.path.append('../insns')
+from insn_add import *
 
 class rvfi_insn_check(Elaboratable):
     def __init__(self, RISCV_FORMAL_ILEN=32, RISCV_FORMAL_XLEN=32):
@@ -200,7 +203,3 @@ class rvfi_insn_check(Elaboratable):
                 m.d.comb += Assert(spec_trap == trap)
         
         return m
-
-test = rvfi_insn_check()
-test.ports()
-test.elaborate(platform=None)