diff --git a/package.json b/package.json
index 1d51aad..858738b 100644
--- a/package.json
+++ b/package.json
@@ -3,7 +3,7 @@
   "name": "mysql-queue",
   "description": "A lite job queue for Node.js",
   "author": "lilac <hi@rabain.com>",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
diff --git a/src/db/index.ts b/src/db/index.ts
index fb2a0ea..6185eaa 100644
--- a/src/db/index.ts
+++ b/src/db/index.ts
@@ -4,19 +4,19 @@ import mysql from "mysql2/promise";
 import path from "node:path";
 import * as schema from "./schema";
 
-export type Database = MySql2Database<typeof schema>;
+export type Database = MySql2Database<Record<string, unknown>>;
 
 export const affectedRows = (rawResult: MySqlRawQueryResult) => {
     return rawResult[0].affectedRows
 };
 
-export async function connect(url: string) {
-    const connection = await mysql.createConnection(url);
+export function connect(url: string) {
+    const connection = mysql.createPool(url);
     const db = drizzle(connection, { schema, mode: 'default' });
     return db;
 }
 
-export function migrateDB(db: MySql2Database<Record<string, unknown>>) {
+export function migrateDB(db: Database) {
     return migrate(db, {
         migrationsFolder: path.join(import.meta.dirname, '../drizzle')
     });
diff --git a/src/test.ts b/src/test.ts
index 5d5fb1f..83a7323 100644
--- a/src/test.ts
+++ b/src/test.ts
@@ -5,7 +5,7 @@ const defaultUrl = env['DATABASE_URL'] ?? 'mysql://root:root@localhost:3306/queu
 
 
 async function prepareDB(url?: string) {
-  const db = await connect(url ?? defaultUrl);
+  const db = connect(url ?? defaultUrl);
   await migrateDB(db);
   return db;
 }