diff --git a/javascript/script.js b/javascript/script.js
index b8f474d04..1ea20e6ca 100644
--- a/javascript/script.js
+++ b/javascript/script.js
@@ -1,5 +1,5 @@
 async function sleep(ms) {
-  return new Promise((resolve) => setTimeout(resolve, ms)); // eslint-disable-line no-promise-executor-return
+  return new Promise((resolve) => { setTimeout(resolve, ms); });
 }
 
 function gradioApp() {
@@ -10,9 +10,14 @@ function gradioApp() {
 }
 
 function logFn(func) {
-  return async function () { // eslint-disable-line func-names
+  return async (...args) => {
     const t0 = performance.now();
-    const returnValue = func(...arguments);
+    let returnValue;
+    if (func.constructor.name.toLowerCase() === 'asyncfunction') {
+      returnValue = await func(...args);
+    } else {
+      returnValue = func(...args);
+    }
     const t1 = performance.now();
     log(func.name, `time=${Math.round(t1 - t0)}`);
     return returnValue;