diff --git a/packages/contract-cli/src/generator/get-socket-auth-resolver.js b/packages/contract-cli/src/generator/get-socket-auth-resolver.js
index fbdd27db5d6efaeded045622d853bc8f2c9ac4de..42b39ff92653242e9264e82cf9837df6f05149f2 100644
--- a/packages/contract-cli/src/generator/get-socket-auth-resolver.js
+++ b/packages/contract-cli/src/generator/get-socket-auth-resolver.js
@@ -114,9 +114,11 @@ function getSocketAuthResolver(config, resolverDir, sourceType = null, fileExt =
       return chainPromises(
         names.map((n, i) => (
           Promise.resolve(processFileToContract(n, files[i], sourceType))
-        ))
+        )),
+        true
       )
     })
+    .then(contract => ({[AUTH_NAME]: contract}))
 }
 
 
diff --git a/packages/contract-cli/tests/socket.test.js b/packages/contract-cli/tests/socket.test.js
index 253e1bf7a774d082c534f56c9aff2322df83aa0e..2e083ad75676abeec7d1ff78dbc9c8b995b967c0 100644
--- a/packages/contract-cli/tests/socket.test.js
+++ b/packages/contract-cli/tests/socket.test.js
@@ -37,7 +37,7 @@ test.cb(`It should able to return a list of socket auth files`, t => {
     })
 })
 
-test.cb.only(`It should able to generate partial contract for socket auth`, t => {
+test.cb(`It should able to generate partial contract for socket auth`, t => {
   t.plan(1)
 
   getSocketAuthResolver(t.context.config, resolverDir)