diff --git a/crates/xtrain-train/tests/grad_accum.rs b/crates/xtrain-train/tests/grad_accum.rs
index e081c1c..d39e390 100644
--- a/crates/xtrain-train/tests/grad_accum.rs
+++ b/crates/xtrain-train/tests/grad_accum.rs
@@ -20,7 +20,6 @@
 #![cfg(not(no_cuda))]
 
 use xtrain_autodiff::ops;
-use xtrain_autodiff::tape::Var;
 use xtrain_cuda::device;
 use xtrain_model::{Config, TinyTransformer, batched_ids_tensor};
 use xtrain_tensor::Device;