1 files changed, 2 insertions, 5 deletions
diff --git a/src/main.rs b/src/main.rs
index d1e4562..de3da62 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,7 +5,6 @@ use anyhow::Result;
 use clap::{Arg, ArgAction, Command, ValueHint, command, value_parser};
 use clap_complete::{Generator, Shell, generate};
 use console::style;
-use rusqlite::Connection;
 use std::{
     path::PathBuf,
     sync::{
@@ -14,8 +13,6 @@ use std::{
     },
 };
 
-use crate::db::Database;
-
 fn build_cli() -> Command {
     command!()
         .arg(
@@ -93,12 +90,12 @@ fn main() -> Result<()> {
         r.store(false, Ordering::SeqCst);
     })?;
 
-    let conn = Connection::new(args.get_one::<PathBuf>("db"))?;
+    let conn = db::init_connection(args.get_one::<PathBuf>("db"))?;
 
     let path = args.get_one::<PathBuf>("path");
 
     if path.is_none() && !args.get_flag("clean") && !args.get_flag("doit") {
-        let count = conn.get_toencode_number()?;
+        let count = db::get_toencode_number(&conn)?;
         println!("Files to reencode:\t{}", style(count).green());
         return Ok(());
     }