diff --git a/packages/mobx-state-tree/__tests__/core/action.test.ts b/packages/mobx-state-tree/__tests__/core/action.test.ts
index 5897c344f..c291b3578 100644
--- a/packages/mobx-state-tree/__tests__/core/action.test.ts
+++ b/packages/mobx-state-tree/__tests__/core/action.test.ts
@@ -10,9 +10,9 @@ import {
     cast,
     IMiddlewareEvent,
     ISerializedActionCall,
-    Instance
+    Instance,
+    mstRunInAction
 } from "../../src"
-import { mstRunInAction } from "../../src/internal"
 
 /// Simple action replay and invocation
 const Task = types