diff --git a/ext/MollyCUDAExt.jl b/ext/MollyCUDAExt.jl
index 5a14a479..a6a0b9c8 100644
--- a/ext/MollyCUDAExt.jl
+++ b/ext/MollyCUDAExt.jl
@@ -2,6 +2,7 @@ module MollyCUDAExt
 
 using Molly
 using CUDA
+using ChainRulesCore
 
 # CUDA specific calls for Molly
 @non_differentiable CUDA.zeros(args...)