diff --git a/Cargo.toml b/Cargo.toml
index 02dee77e..83aae6f9 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -114,13 +114,3 @@ runtime-actix-rustls = ["sqlx/runtime-actix-rustls", "runtime-actix"]
 runtime-tokio = []
 runtime-tokio-native-tls = ["sqlx/runtime-tokio-native-tls", "runtime-tokio"]
 runtime-tokio-rustls = ["sqlx/runtime-tokio-rustls", "runtime-tokio"]
-
-[[test]]
-name = "sqlite-basic"
-path = "tests/basic.rs"
-required-features = ["sqlx-sqlite"]
-
-[[test]]
-name = "postgres"
-path = "tests/pg_tests.rs"
-required-features = ["sqlx-postgres"]
diff --git a/tests/basic.rs b/tests/basic.rs
index 20256ad4..5efe6ea2 100644
--- a/tests/basic.rs
+++ b/tests/basic.rs
@@ -3,9 +3,9 @@ use sea_orm::{entity::*, error::*, sea_query, tests_cfg::*, DbBackend, DbConn, S
 mod setup;
 
 // cargo test --test basic -- --nocapture
-#[cfg_attr(feature = "runtime-async-std", async_std::main)]
-#[cfg_attr(feature = "runtime-actix", actix_rt::main)]
-#[cfg_attr(feature = "runtime-tokio", tokio::main)]
+#[cfg_attr(feature = "runtime-async-std", async_std::test)]
+#[cfg_attr(feature = "runtime-actix", actix_rt::test)]
+#[cfg_attr(feature = "runtime-tokio", tokio::test)]
 #[cfg(feature = "sqlx-sqlite")]
 async fn main() {
     let db: DbConn = setup::setup().await;
diff --git a/tests/pg_tests.rs b/tests/pg_tests.rs
index c370112d..ca0dfa82 100644
--- a/tests/pg_tests.rs
+++ b/tests/pg_tests.rs
@@ -9,9 +9,9 @@ pub use common::bakery_chain::*;
 use sea_query::{ColumnDef, TableCreateStatement};
 
 // cargo test --test pg_tests -- --nocapture
-#[cfg_attr(feature = "runtime-async-std", async_std::main)]
-#[cfg_attr(feature = "runtime-actix", actix_rt::main)]
-#[cfg_attr(feature = "runtime-tokio", tokio::main)]
+#[cfg_attr(feature = "runtime-async-std", async_std::test)]
+#[cfg_attr(feature = "runtime-actix", actix_rt::test)]
+#[cfg_attr(feature = "runtime-tokio", tokio::test)]
 #[cfg(feature = "sqlx-postgres")]
 async fn main() {
     let base_url = "postgres://root:root@localhost";