diff --git a/public/scripts/extensions/shared.js b/public/scripts/extensions/shared.js
index 9c39a18d1..a2c15ba41 100644
--- a/public/scripts/extensions/shared.js
+++ b/public/scripts/extensions/shared.js
@@ -251,7 +251,7 @@ export async function countWebLlmTokens(text) {
         return response;
     } catch (error) {
         // Fallback to using current model's tokenizer
-        return getTokenCountAsync(text);
+        return await getTokenCountAsync(text);
     }
 }