diff --git a/server.js b/server.js
index 1cb676993..290695f63 100644
--- a/server.js
+++ b/server.js
@@ -3808,7 +3808,7 @@ async function sendAI21Request(request, response) {
 
 }
 
-app.post("/tokenize_ai21", jsonParser, function (request, response_tokenize_ai21) {
+app.post("/tokenize_ai21", jsonParser, async function (request, response_tokenize_ai21) {
     if (!request.body) return response_tokenize_ai21.sendStatus(400);
     const options = {
         method: 'POST',
@@ -3820,10 +3820,14 @@ app.post("/tokenize_ai21", jsonParser, function (request, response_tokenize_ai21
         body: JSON.stringify({ text: request.body[0].content })
     };
 
-    fetch('https://api.ai21.com/studio/v1/tokenize', options)
-        .then(response => response.json())
-        .then(response => response_tokenize_ai21.send({ "token_count": response.tokens.length }))
-        .catch(err => console.error(err));
+    try {
+        const response = await fetch('https://api.ai21.com/studio/v1/tokenize', options);
+        const data = await response.json();
+        return response_tokenize_ai21.send({ "token_count": data?.tokens?.length || 0 });
+    } catch (err) {
+        console.error(err);
+        return response_tokenize_ai21.send({ "token_count": 0 });
+    }
 });
 
 app.post("/save_preset", jsonParser, function (request, response) {