diff --git a/src/main.rs b/src/main.rs
index bcf3d92..612a723 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -79,7 +79,7 @@ mod tests {
 
         pub async fn generate_db_name() -> String {
             let db_name =
-                get_database_name().unwrap() + &"_" + &uuid::Uuid::new_v4().to_string()[..LIMIT];
+                get_database_name().await.unwrap() + &"_" + &uuid::Uuid::new_v4().to_string()[..LIMIT];
             db_name
         }
 
@@ -110,11 +110,8 @@ mod tests {
             Ok(())
         }
 
-        pub fn get_database_name() -> Result<String, Box<dyn std::error::Error>> {
-            let rt = tokio::runtime::Runtime::new().unwrap();
-
-            // Block on the async function to get the result
-            let database_url  = rt.block_on(icarus_envy::environment::get_db_url());
+        pub async fn get_database_name() -> Result<String, Box<dyn std::error::Error>> {
+            let database_url = icarus_envy::environment::get_db_url().await;
 
             let parsed_url = url::Url::parse(&database_url)?;
             if parsed_url.scheme() == "postgres" || parsed_url.scheme() == "postgresql" {