diff --git a/lib/queries.js b/lib/queries.js
index 03b4759f..3bfc27fe 100644
--- a/lib/queries.js
+++ b/lib/queries.js
@@ -6,7 +6,7 @@ const POSTGRESQL = 'postgresql';
 const MYSQL = 'mysql';
 
 export function getDatabase() {
-  return process.env.DATABASE_URL.split(':')[0];
+  return process.env.DATABASE_TYPE || process.env.DATABASE_URL.split(':')[0];
 }
 
 export async function getWebsiteById(website_id) {