diff --git a/src/lib.rs b/src/lib.rs
index b057a89..f895d80 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,2 +1,4 @@
-pub mod query;
+mod query;
 pub mod response;
+
+pub use query::query_server;
diff --git a/src/main.rs b/src/main.rs
index c465039..3b9f39c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,9 +1,9 @@
 use anyhow::Result;
 
-use departure::query;
+use departure::query_server;
 
 fn main() -> Result<()> {
-    let response = query::query_server()?;
+    let response = query_server()?;
     for dep in response.departures {
         println!("{dep}");
     }