1 files changed, 5 insertions, 12 deletions
diff --git a/src/kona-cli.rs b/src/kona-cli.rs
index 04adfc8..74d7c7f 100644
--- a/src/kona-cli.rs
+++ b/src/kona-cli.rs
@@ -15,18 +15,11 @@
 // You should have received a copy of the GNU Affero General Public
 // License along with Kona. If not, see <http://www.gnu.org/licenses/>.
 
-#[macro_use]
-extern crate anyhow;
-extern crate dirs;
-#[macro_use]
-extern crate rusqlite;
-
-mod database;
-
-use database::TagDatabase;
 use std::env;
 use std::path::Path;
 
+use kona::TagDatabase;
+
 fn main() {
     let mut args: Vec<String> = env::args().collect();
     if args.len() < 2 || args.iter().any(|arg| arg == "-h" || arg == "--help") {
@@ -50,7 +43,7 @@ fn main() {
         }
 
         args.remove(i);
-        TagDatabase::new(args.remove(i + 1))
+        TagDatabase::new(args.remove(i))
     } else if let Ok(path) = std::env::var("TB_PATH") {
         TagDatabase::new(path)
     } else {
@@ -100,9 +93,9 @@ fn main() {
         }
         "query" => {
             let query = if args.len() < 3 {
-                database::parse_query("").unwrap()
+                kona::parse_query("").unwrap()
             } else {
-                database::parse_query(&args[2]).unwrap()
+                kona::parse_query(&args[2]).unwrap()
             };
             for file in tb.query(&query, None).unwrap() {
                 println!("{}", file.path);