diff --git a/src/YaoHIR.jl b/src/YaoHIR.jl
index ef33463..42e4cf0 100644
--- a/src/YaoHIR.jl
+++ b/src/YaoHIR.jl
@@ -10,6 +10,7 @@ export GenericRoutine, Routine,
 using MLStyle
 using Expronicon
 using Core: CodeInfo
+using Core.Compiler: IRCode
 
 include("types.jl")
 include("intrinsic.jl")
diff --git a/src/types.jl b/src/types.jl
index 627e59a..b352ab3 100644
--- a/src/types.jl
+++ b/src/types.jl
@@ -104,7 +104,7 @@ end
 Base.adjoint(x::Ctrl) = Ctrl(adjoint(x.gate), x.ctrl)
 
 struct BlockIR
-    parent::CodeInfo
+    parent::IRCode
     nqubits::Int
     circuit::Chain
 end