diff --git a/actions-languageserver/src/utils/timer.ts b/actions-languageserver/src/utils/timer.ts
index 0278a11..f9dc71e 100644
--- a/actions-languageserver/src/utils/timer.ts
+++ b/actions-languageserver/src/utils/timer.ts
@@ -1,8 +1,12 @@
 import {log} from "@github/actions-languageservice/log";
 
-export function timeOperation<T>(name: string, f: () => T): T {
+export async function timeOperation<T>(name: string, f: () => T): Promise<T> {
   const start = Date.now();
   const result = f();
+  if (result instanceof Promise) {
+    await result;
+  }
+
   const end = Date.now();
 
   log(`${name} took ${end - start}ms`);