diff --git a/models/src/utils/token_stream.rs b/models/src/utils/token_stream.rs
index d74bf3e..0435611 100644
--- a/models/src/utils/token_stream.rs
+++ b/models/src/utils/token_stream.rs
@@ -1,4 +1,4 @@
-use candle_core::Result;
+use candle::Result;
 
 /// This is a wrapper around a tokenizer to ensure that tokens can be returned to the user in a
 /// streaming way rather than having to wait for the full decoding.