diff --git a/packages/overmind/src/statemachine.test.ts b/packages/overmind/src/statemachine.test.ts
index 7223b26a..0b401323 100644
--- a/packages/overmind/src/statemachine.test.ts
+++ b/packages/overmind/src/statemachine.test.ts
@@ -328,11 +328,13 @@ describe('Statemachine', () => {
         foo: 'bar',
       },
     })
-
+    type Context = IContext<{
+      state: typeof state
+    }>
     const config = {
       state,
       actions: {
-        changeFoo({ state }) {
+        changeFoo({ state }: Context) {
           state.obj.foo = 'bar2'
         },
       },