diff --git a/src/index.test.ts b/src/index.test.ts
index d873538c..c5d3c987 100644
--- a/src/index.test.ts
+++ b/src/index.test.ts
@@ -1,4 +1,4 @@
-import type { PlayFunctionContext } from '@storybook/types';
+import type { PlayFunctionContext } from 'storybook/internal/types';
 import type { Component, ComponentProps, Snippet } from 'svelte';
 import type { EmptyObject, Primitive } from 'type-fest';
 import { describe, expectTypeOf, it } from 'vitest';
diff --git a/src/runtime/Story.svelte b/src/runtime/Story.svelte
index 44b50e39..0d91ba1e 100644
--- a/src/runtime/Story.svelte
+++ b/src/runtime/Story.svelte
@@ -5,7 +5,7 @@
   lang="ts"
   generics="const TOverrideArgs extends Args, const TCmp extends Cmp, TMeta extends Meta<TCmp>"
 >
-  import type { Args } from '@storybook/types';
+  import type { Args } from 'storybook/internal/types';
   import type { Snippet } from 'svelte';
 
   import { useStoriesExtractor } from '#runtime/contexts/extractor.svelte';
diff --git a/src/runtime/contexts/extractor.svelte.ts b/src/runtime/contexts/extractor.svelte.ts
index 36e33d33..a2960b4a 100644
--- a/src/runtime/contexts/extractor.svelte.ts
+++ b/src/runtime/contexts/extractor.svelte.ts
@@ -1,4 +1,4 @@
-import type { Args } from '@storybook/types';
+import type { Args } from 'storybook/internal/types';
 import { getContext, hasContext, setContext, type ComponentProps } from 'svelte';
 
 import { storyNameToExportName } from '#utils/identifier-utils';
diff --git a/src/runtime/contexts/renderer.svelte.ts b/src/runtime/contexts/renderer.svelte.ts
index 6648ae97..e303bfbe 100644
--- a/src/runtime/contexts/renderer.svelte.ts
+++ b/src/runtime/contexts/renderer.svelte.ts
@@ -1,7 +1,7 @@
 import { getContext, hasContext, setContext } from 'svelte';
 
 import type { Cmp, Meta, StoryAnnotations, StoryContext } from '#types';
-import type { Args } from '@storybook/types';
+import type { Args } from 'storybook/internal/types';
 
 const CONTEXT_KEY = 'storybook-story-renderer-context';
 
diff --git a/src/runtime/contexts/template.svelte.ts b/src/runtime/contexts/template.svelte.ts
index aaee9ff7..1c736e6f 100644
--- a/src/runtime/contexts/template.svelte.ts
+++ b/src/runtime/contexts/template.svelte.ts
@@ -1,4 +1,4 @@
-import type { Args } from '@storybook/types';
+import type { Args } from 'storybook/internal/types';
 import { getContext, hasContext, setContext, type ComponentProps } from 'svelte';
 
 import type { Cmp, Meta, StoryCmp } from '#types';
diff --git a/src/types.test.ts b/src/types.test.ts
index d0e98637..0e78dd51 100644
--- a/src/types.test.ts
+++ b/src/types.test.ts
@@ -1,4 +1,4 @@
-import type { PlayFunctionContext } from '@storybook/types';
+import type { PlayFunctionContext } from 'storybook/internal/types';
 import type { Component, ComponentProps } from 'svelte';
 import { describe, expectTypeOf, it } from 'vitest';