diff --git a/src/Enzyme.jl b/src/Enzyme.jl
index 4007029609..1662a243bc 100644
--- a/src/Enzyme.jl
+++ b/src/Enzyme.jl
@@ -13,7 +13,7 @@ export autodiff, jacobian, gradient, gradient!
 export markType, batch_size, onehot, chunkedonehot
 
 using LinearAlgebra
-import EnzymeCore: ReverseMode, ForwardMode, Annotation 
+import EnzymeCore: ReverseMode, ForwardMode, Annotation, Mode
 
 # Independent code, must be loaded before "compiler.jl"
 include("pmap.jl")