diff --git a/apps/server/src/services/cls.ts b/apps/server/src/services/cls.ts
index 5fae8afbd..5835385be 100644
--- a/apps/server/src/services/cls.ts
+++ b/apps/server/src/services/cls.ts
@@ -3,8 +3,8 @@ import { getContext } from "@triliumnext/core/src/services/context";
 
 type Callback = (...args: any[]) => any;
 
-function init(callback: () => void) {
-    getContext().init(callback);
+function init<T>(callback: () => T) {
+    return getContext().init(callback);
 }
 
 function wrap(callback: Callback) {