diff --git a/hfd-cli/Cargo.toml b/hfd-cli/Cargo.toml
index a982674..d34b732 100644
--- a/hfd-cli/Cargo.toml
+++ b/hfd-cli/Cargo.toml
@@ -1,11 +1,12 @@
 [package]
 name = "hfd-cli"
-version = "0.1.4"
+version = "0.1.5"
 edition = "2021"
 
 [dependencies]
 tokio = { version = "1.37.0", features = ["full"] }
 clap = { version= "4.5.4", features=["derive"] }
+tokio-stream = "0.1.15"
 libhfd = { path = "../hfd" }
 
 [[bin]]
diff --git a/hfd-cli/src/cli.rs b/hfd-cli/src/cli.rs
index e281490..e854245 100644
--- a/hfd-cli/src/cli.rs
+++ b/hfd-cli/src/cli.rs
@@ -1,11 +1,12 @@
 use tokio;
+use tokio_stream::StreamExt;
 
 #[tokio::main]
 async fn main() {
     let api = libhfd::api::tokio::Api::new().unwrap();
 
     let _filename = api
-        .model("ByteDance/Hyper-SD".to_string())
+        .dataset("ByteDance/Hyper-SD".to_string())
         .get("Hyper-SDXL-8steps-lora.safetensors")
         .await
         .unwrap();