diff --git a/packages/restate-sdk/src/common_api.ts b/packages/restate-sdk/src/common_api.ts
index 306a9e31..07429159 100644
--- a/packages/restate-sdk/src/common_api.ts
+++ b/packages/restate-sdk/src/common_api.ts
@@ -21,6 +21,8 @@ export type {
   InvocationId,
   InvocationHandle,
   InvocationPromise,
+  TypedState,
+  UntypedState,
 } from "./context.js";
 export { InvocationIdParser, CombineablePromise } from "./context.js";