diff --git a/index.js b/index.js
index e526b20..041f5e0 100644
--- a/index.js
+++ b/index.js
@@ -114,14 +114,26 @@ async function postCode() {
   let api;
   if (token) {
     api = new ScreepsAPI(login_arguments);
+    const response = await api.code.set(branch, files_to_push);
+    core.info(JSON.stringify(response, null, 2));
   } else {
     core.info(`Logging into as user ${username}`);
-    api = new ScreepsAPI(login_arguments);
-    await api.auth(username, password);
+    const response = Promise.resolve()
+      .then(() => api.auth(auth.email, auth.password))
+      .then(() => api.socket.connect())
+      .then(() => {
+        return api.code.set(branch, code); // Set the code on the specified branch
+      })
+      .then(() => {
+        console.log("Code set successfully");
+        // Additional actions after setting the code, if necessary
+      })
+      .catch((err) => {
+        console.error("Error:", err);
+      });
     core.info("Authorized!");
+    core.info(JSON.stringify(response, null, 2));
   }
-  const response = await api.code.set(branch, files_to_push);
-  core.info(JSON.stringify(response, null, 2));
 }
 
 postCode();