diff --git a/whatsapp-gateway/baileys-client.js b/whatsapp-gateway/baileys-client.js
index 1928428..5ff5ff3 100644
--- a/whatsapp-gateway/baileys-client.js
+++ b/whatsapp-gateway/baileys-client.js
@@ -455,11 +455,17 @@ function getActiveSessions() {
     return Array.from(sessions.keys());
 }
 
+function isSessionReady(session_key) {
+    const sock = sessions.get(session_key);
+    return !!(sock && sock.user);
+}
+
 module.exports = {
     startSession,
     disconnectSession,
     sendMessage,
     getActiveSessions,
-    checkContact
+    checkContact,
+    isSessionReady
 };
 
diff --git a/whatsapp-gateway/package.json b/whatsapp-gateway/package.json
index b6e640c..9f720cf 100644
--- a/whatsapp-gateway/package.json
+++ b/whatsapp-gateway/package.json
@@ -7,6 +7,7 @@
     "start": "node server.js"
   },
   "dependencies": {
+    "@whiskeysockets/baileys": "^6.7.0",
     "whatsapp-web.js": "^1.26.0",
     "puppeteer": "^21.0.0",
     "qrcode": "^1.5.3",
diff --git a/whatsapp-gateway/server.js b/whatsapp-gateway/server.js
index acc694c..f28dec3 100644
--- a/whatsapp-gateway/server.js
+++ b/whatsapp-gateway/server.js
@@ -20,7 +20,7 @@ for (const p of envPaths) {
 
 const express = require('express');
 const cors = require('cors');
-const { startSession, disconnectSession, sendMessage, getActiveSessions, checkContact, isSessionReady } = require('./puppeteer-client');
+const { startSession, disconnectSession, sendMessage, getActiveSessions, checkContact, isSessionReady } = require('./baileys-client');
 
 const app = express();
 app.use(cors());