diff --git a/CHANGELOG.md b/CHANGELOG.md
index fe3dc17..f7f36c9 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,6 @@
+## v0.1.2 (2023-12-12)
+* Fixed JWT wrapper return types
+
 ## v0.1.1 (2023-12-12)
 * Added date as Carbon type
 * Pass Cipher payload as JWT wrapper argument
diff --git a/src/Indoctrination/Context.php b/src/Indoctrination/Context.php
index ee8d09c..7e36eee 100644
--- a/src/Indoctrination/Context.php
+++ b/src/Indoctrination/Context.php
@@ -245,6 +245,12 @@ public function clearCache(): void
 
     /**
      * Transaction with RLS
+     *
+     * @template TReturn
+     * @template TPayload of CipherPayload
+     * @param TPayload $payload
+     * @param Closure(EntityManager, TPayload):TReturn $callback
+     * @return TReturn
      */
     public function withJwt(
         CipherPayload $payload,
@@ -272,6 +278,10 @@ public function withJwt(
 
     /**
      * Bypass RLS
+     *
+     * @template TReturn
+     * @param Closure(EntityManager):TReturn $callback
+     * @return TReturn
      */
     public function bypassJwt(
         Closure $callback,