diff --git a/Cargo.toml b/Cargo.toml
index d799d71..7fc5dde 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -68,9 +68,8 @@ netlink-packet-route = "0.29"
 netlink-sys = "0.8"
 
 # ML (optional)
-candle-core = { version = "0.10", optional = true }
-candle-nn = { version = "0.3", optional = true }
-
+candle-core = { version = "0.3", optional = true }
+candle-nn = { version = "0.10", optional = true }
 [features]
 default = []
 ml = ["candle-core", "candle-nn"]