diff --git a/packages/astro/src/default/env-default.d.ts b/packages/astro/src/default/env-default.d.ts
index f61727d18..8507091db 100644
--- a/packages/astro/src/default/env-default.d.ts
+++ b/packages/astro/src/default/env-default.d.ts
@@ -13,5 +13,7 @@ declare module 'tutorialkit:override-components' {
   export { topBar as TopBar };
 }
 
+declare module 'tutorialkit:astro-swap-functions';
+
 declare const __ENTERPRISE__: boolean;
 declare const __WC_CONFIG__: WebContainerConfig | undefined;
diff --git a/packages/astro/src/default/layouts/Layout.astro b/packages/astro/src/default/layouts/Layout.astro
index b77544c81..4b8ceb3a0 100644
--- a/packages/astro/src/default/layouts/Layout.astro
+++ b/packages/astro/src/default/layouts/Layout.astro
@@ -38,7 +38,7 @@ const baseURL = import.meta.env.BASE_URL;
       }
     </script>
     <script>
-      import * as builtInSwap from 'node_modules/astro/dist/transitions/swap-functions';
+      import * as builtInSwap from 'tutorialkit:astro-swap-functions';
 
       declare global {
         function setTutorialKitTheme(): void;
diff --git a/packages/astro/src/index.ts b/packages/astro/src/index.ts
index 02bd01f0d..349d75447 100644
--- a/packages/astro/src/index.ts
+++ b/packages/astro/src/index.ts
@@ -4,6 +4,7 @@ import { fileURLToPath } from 'node:url';
 import { extraIntegrations } from './integrations.js';
 import { updateMarkdownConfig } from './remark/index.js';
 import { tutorialkitCore } from './vite-plugins/core.js';
+import { astroSwapFunctions } from './vite-plugins/astro-swap-functions.js';
 import { userlandCSS, watchUserlandCSS } from './vite-plugins/css.js';
 import { tutorialkitStore } from './vite-plugins/store.js';
 import { overrideComponents, type OverrideComponentsOptions } from './vite-plugins/override-components.js';
@@ -108,6 +109,7 @@ export default function createPlugin({
               tutorialkitStore,
               tutorialkitCore,
               overrideComponents({ components, defaultRoutes: !!defaultRoutes }),
+              astroSwapFunctions(),
               process.env.TUTORIALKIT_DEV ? (await import('vite-plugin-inspect')).default() : null,
             ],
           },
diff --git a/packages/astro/src/vite-plugins/astro-swap-functions.ts b/packages/astro/src/vite-plugins/astro-swap-functions.ts
new file mode 100644
index 000000000..d67ec3c95
--- /dev/null
+++ b/packages/astro/src/vite-plugins/astro-swap-functions.ts
@@ -0,0 +1,35 @@
+/**
+ * A plugin for importing Astro's private 'transitions/swap-functions.js' module.
+ * This is temporary solution and can break at any time when end-user's update their `astro` version.
+ */
+import { createRequire } from 'node:module';
+import { resolve, sep } from 'node:path';
+import type { VitePlugin } from '../types.js';
+
+const require = createRequire(import.meta.url);
+const astroDist = resolve(require.resolve('astro/package.json'), '..');
+const swapFunctionEntry = resolve(astroDist, 'dist/transitions/swap-functions.js').replaceAll(sep, '/');
+
+const virtualModuleId = 'tutorialkit:astro-swap-functions';
+const resolvedId = `\0${virtualModuleId}`;
+
+export function astroSwapFunctions(): VitePlugin {
+  return {
+    name: 'tutorialkit-astro-swap-functions-plugin',
+    resolveId(id) {
+      if (id === virtualModuleId) {
+        return resolvedId;
+      }
+
+      return undefined;
+    },
+
+    async load(id) {
+      if (id === resolvedId) {
+        return `export * from '${swapFunctionEntry}';`;
+      }
+
+      return undefined;
+    },
+  };
+}