diff --git a/src/main.rs b/src/main.rs
index 1d61d3e..0f53016 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,6 @@
-use crate::ask;
+extern crate baichat_rs;
+
+use baichat_rs::ask;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {