diff --git a/jest.setup.ts b/jest.setup.ts
index a9ff61fa4..161bc0b20 100644
--- a/jest.setup.ts
+++ b/jest.setup.ts
@@ -1,11 +1,19 @@
-import path from 'path';
 import dotenv from 'dotenv';
+import { existsSync } from 'fs';
+import { resolve } from 'path';
 import prettyFormat from 'pretty-format';
+import yargs from 'yargs';
+
+const envFile = existsSync(resolve(__dirname, '.env.test')) ? '.env.test' : '.env';
 
 dotenv.config({
-  path: path.resolve(__dirname, process.env.ENV_FILE || '.env'),
+  path: resolve(__dirname, envFile),
 });
 
+if (yargs.argv.dbDialect) {
+  process.env.DB_DIALECT = yargs.argv.dbDialect as any;
+}
+
 global['prettyFormat'] = prettyFormat;
 
 jest.setTimeout(300000);