diff --git a/src/LegacySocket/socket.ts b/src/LegacySocket/socket.ts
index f16c2d8..d9fa528 100644
--- a/src/LegacySocket/socket.ts
+++ b/src/LegacySocket/socket.ts
@@ -363,7 +363,7 @@ export const makeSocket = ({
     })
 
 	return {
-        type: 'legacy',
+        type: 'legacy' as 'legacy',
         ws,
         updateKeys: (info: { encKey: Buffer, macKey: Buffer }) => authInfo = info,
         waitForSocketOpen,
diff --git a/src/Socket/socket.ts b/src/Socket/socket.ts
index 2007722..059cb33 100644
--- a/src/Socket/socket.ts
+++ b/src/Socket/socket.ts
@@ -540,7 +540,7 @@ export const makeSocket = ({
     ev.on('creds.update', update => Object.assign(creds, update))
 
 	return {
-        type: 'md',
+        type: 'md' as 'md',
         ws,
         ev,
         authState: {