diff --git a/candle_demo/Cargo.toml b/candle_demo/Cargo.toml
index a66949d..2e3f4b6 100755
--- a/candle_demo/Cargo.toml
+++ b/candle_demo/Cargo.toml
@@ -32,5 +32,5 @@ bindgen_cuda = { version = "0.1.1", optional = true }
 
 [features]
 default = []
-cuda = ["candle/cuda", "candle-nn/cuda", "candle-transformers/cuda", "dep:bindgen_cuda"]
+cuda = ["candle-core/cuda", "candle-nn/cuda", "candle-transformers/cuda", "dep:bindgen_cuda"]