diff --git a/src/main.rs b/src/main.rs
index 89fdc0a..3148662 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -206,11 +206,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" {