diff --git a/src/main.rs b/src/main.rs
index d148b965f2412daee7f108043742dddcd9d193f0..2d0a018905e37666b6d3b6920c19f0ea9cd1795b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -124,9 +124,9 @@ fn main() -> Result<(), io::Error> {
 
 fn read_from_db(config: HashMap<String, String>, mut predictor: SetPredictor) -> SetPredictor {
     let default_password = &String::from("");
-    match config.get("db_password") {
-        Some(db_password) => {
-            let mut client = Client::connect(format!("host=database-postgresql.metager-2 user=metager password={} dbname=postgres",db_password).as_str(), NoTls);
+    match (config.get("db_host"), config.get("db_password")) {
+        (Some(db_host), Some(db_password)) => {
+            let mut client = Client::connect(format!("host={} user=metager password={} dbname=postgres",db_host, db_password).as_str(), NoTls);
             match client {
                 Ok(mut c) => {
                     let mut count = 0;
@@ -152,8 +152,8 @@ fn read_from_db(config: HashMap<String, String>, mut predictor: SetPredictor) ->
                 }
             }
         },
-        None => {
-            println!("No DB password found. Skipping...");
+        _ => {
+            println!("No DB config found. Skipping...");
             predictor
         }
     }