diff --git a/src/types.ts b/src/types.ts
index 0276f0b..320ddf4 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -121,7 +121,12 @@ export type XstateTreeMachine<
 /**
  * @public
  */
-export type AnyXstateTreeMachine = XstateTreeMachine<AnyStateMachine>;
+export type AnyXstateTreeMachine = XstateTreeMachine<
+  AnyStateMachine,
+  any,
+  any,
+  any[]
+>;
 /**
  * @internal
  */
diff --git a/src/xstateTree.spec.tsx b/src/xstateTree.spec.tsx
index 7ab85c8..f8a3dda 100644
--- a/src/xstateTree.spec.tsx
+++ b/src/xstateTree.spec.tsx
@@ -69,6 +69,13 @@ describe("xstate-tree", () => {
     });
 
     const XstateTreeMachine = createXStateTreeMachine(machine, {
+      // Had a type error when adding in actions to this machine which has now been resolved
+      // keeping it for sanity check
+      actions() {
+        return {
+          foo() {},
+        };
+      },
       View: ({ selectors }) => {
         return <p>{selectors.foo}</p>;
       },
diff --git a/xstate-tree.api.md b/xstate-tree.api.md
index 6be27dd..6de2ab7 100644
--- a/xstate-tree.api.md
+++ b/xstate-tree.api.md
@@ -46,7 +46,7 @@ export type AnyRoute = {
 };
 
 // @public (undocumented)
-export type AnyXstateTreeMachine = XstateTreeMachine<AnyStateMachine>;
+export type AnyXstateTreeMachine = XstateTreeMachine<AnyStateMachine, any, any, any[]>;
 
 // @public (undocumented)
 export type ArgumentsForRoute<T> = T extends Route<infer TParams, infer TQuery, any, infer TMeta> ? RouteArguments<TParams, TQuery, TMeta> : never;
@@ -400,8 +400,8 @@ export type XstateTreeMachineStateSchemaV2<TMachine extends AnyStateMachine, TSe
 // src/routing/createRoute/createRoute.ts:285:19 - (ae-forgotten-export) The symbol "MergeRouteTypes" needs to be exported by the entry point index.d.ts
 // src/routing/createRoute/createRoute.ts:285:19 - (ae-forgotten-export) The symbol "ResolveZodType" needs to be exported by the entry point index.d.ts
 // src/routing/createRoute/createRoute.ts:322:9 - (ae-forgotten-export) The symbol "RouteRedirect" needs to be exported by the entry point index.d.ts
-// src/types.ts:137:3 - (ae-incompatible-release-tags) The symbol "canHandleEvent" is marked as @public, but its signature references "CanHandleEvent" which is marked as @internal
-// src/types.ts:138:3 - (ae-incompatible-release-tags) The symbol "inState" is marked as @public, but its signature references "MatchesFrom" which is marked as @internal
+// src/types.ts:142:3 - (ae-incompatible-release-tags) The symbol "canHandleEvent" is marked as @public, but its signature references "CanHandleEvent" which is marked as @internal
+// src/types.ts:143:3 - (ae-incompatible-release-tags) The symbol "inState" is marked as @public, but its signature references "MatchesFrom" which is marked as @internal
 
 // (No @packageDocumentation comment for this package)