diff --git a/src/lib/mina/actions/offchain-contract-tests/ExampleContract.ts b/src/lib/mina/actions/offchain-contract-tests/ExampleContract.ts
index 1e4a48251..33e75b52d 100644
--- a/src/lib/mina/actions/offchain-contract-tests/ExampleContract.ts
+++ b/src/lib/mina/actions/offchain-contract-tests/ExampleContract.ts
@@ -27,7 +27,7 @@ class ExampleContract extends SmartContract {
     offchainState.emptyCommitments();
 
   // o1js memoizes the offchain state by contract address so that this pattern works
-  offchainState: any = offchainState.init(this);
+  offchainState = offchainState.init(this);
 
   @method
   async createAccount(address: PublicKey, amountToMint: UInt64) {
diff --git a/src/lib/mina/actions/offchain-state.ts b/src/lib/mina/actions/offchain-state.ts
index af3b5829d..6b091df29 100644
--- a/src/lib/mina/actions/offchain-state.ts
+++ b/src/lib/mina/actions/offchain-state.ts
@@ -33,7 +33,7 @@ import { ProvableType } from '../../provable/types/provable-intf.js';
 export { OffchainState, OffchainStateCommitments };
 
 // internal API
-export { OffchainField, OffchainMap };
+export { OffchainField, OffchainMap, OffchainStateInstance };
 
 type OffchainStateInstance<
   Config extends { [key: string]: OffchainStateKind }