diff --git a/Makefile b/Makefile
index e8095f6..5c2dd6d 100644
--- a/Makefile
+++ b/Makefile
@@ -6,3 +6,6 @@ build:
 
 publish:
 	npm publish --access public
+
+test:
+	pnpm run test
diff --git a/src/pg-adapter.spec.ts b/src/pg-adapter.spec.ts
index 90b510f..2cbd98a 100644
--- a/src/pg-adapter.spec.ts
+++ b/src/pg-adapter.spec.ts
@@ -22,7 +22,7 @@ const callMockedPgAdapater = async (queries: string[]) => {
   const pgAdapter = new PrismaPg(client as any)
 
   for(const sql of queries) {
-    await pgAdapter['performIO']({ sql, args: [] })
+    await pgAdapter['performIO']({ sql, args: [], argTypes: [] })
   }
 
   return query
diff --git a/src/pg-adapter.ts b/src/pg-adapter.ts
index 93b283c..ac1fda2 100644
--- a/src/pg-adapter.ts
+++ b/src/pg-adapter.ts
@@ -239,8 +239,8 @@ class PgTransaction extends PgQueryable<TransactionClient> implements Transactio
   }
 }
 
-class PgTransactionContext extends PgQueryable<pg.PoolClient> implements TransactionContext {
-  constructor(readonly conn: pg.PoolClient) {
+class PgTransactionContext extends PgQueryable<TransactionClient> implements TransactionContext {
+  constructor(readonly conn: TransactionClient) {
     super(conn)
   }