diff --git a/src/compile/wrapModule.ts b/src/compile/wrapModule.ts
index d4f8198bc0..05d33f7fd4 100644
--- a/src/compile/wrapModule.ts
+++ b/src/compile/wrapModule.ts
@@ -41,6 +41,12 @@ export default function wrapModule(
 	throw new Error(`options.format is invalid (must be ${list(Object.keys(wrappers))})`);
 }
 
+function editSource(source, sveltePath) {
+	return source === 'svelte' || source.startsWith('svelte/')
+		? source.replace('svelte', sveltePath)
+		: source;
+}
+
 function esm(
 	code: string,
 	name: string,
@@ -60,7 +66,7 @@ function esm(
 	const importBlock = imports.length > 0 && (
 		imports
 			.map((declaration: Node) => {
-				const import_source = declaration.source.value === 'svelte' ? sveltePath : declaration.source.value;
+				const import_source = editSource(declaration.source.value, sveltePath);
 
 				return (
 					source.slice(declaration.start, declaration.source.start) +
@@ -117,9 +123,7 @@ function cjs(
 			lhs = `{ ${properties.join(', ')} }`;
 		}
 
-		const source = node.source.value === 'svelte'
-			? sveltePath
-			: node.source.value;
+		const source = editSource(node.source.value, sveltePath);
 
 		return `const ${lhs} = require("${source}");`
 	});
diff --git a/test/runtime/samples/context-api/Tab.html b/test/runtime/samples/context-api/Tab.html
new file mode 100644
index 0000000000..720d5da9de
--- /dev/null
+++ b/test/runtime/samples/context-api/Tab.html
@@ -0,0 +1,17 @@
+<script>
+	import { getContext, onDestroy } from 'svelte';
+	import { TABS } from './Tabs.html';
+
+	const tab = {};
+	const { registerTab, unregisterTab, selectTab, selectedTab } = getContext(TABS);
+
+	registerTab(tab);
+
+	onDestroy(() => {
+		unregisterTab(tab);
+	});
+</script>
+
+<button class:selected={$selectedTab} on:click="{() => selectTab(tab)}">
+	<slot></slot>
+</button>
\ No newline at end of file
diff --git a/test/runtime/samples/context-api/TabList.html b/test/runtime/samples/context-api/TabList.html
new file mode 100644
index 0000000000..08c86216ae
--- /dev/null
+++ b/test/runtime/samples/context-api/TabList.html
@@ -0,0 +1,3 @@
+<div class="tab-list">
+	<slot></slot>
+</div>
\ No newline at end of file
diff --git a/test/runtime/samples/context-api/TabPanel.html b/test/runtime/samples/context-api/TabPanel.html
new file mode 100644
index 0000000000..7802ffac26
--- /dev/null
+++ b/test/runtime/samples/context-api/TabPanel.html
@@ -0,0 +1,16 @@
+<script>
+	import { getContext, onDestroy } from 'svelte';
+	import { TABS } from './Tabs.html';
+
+	const panel = {};
+	const { registerPanel, unregisterPanel, selectedPanel } = getContext(TABS);
+	registerPanel(panel);
+
+	onDestroy(() => {
+		unregisterPanel(panel);
+	});
+</script>
+
+{#if $selected === panel}
+	<slot></slot>
+{/if}
\ No newline at end of file
diff --git a/test/runtime/samples/context-api/Tabs.html b/test/runtime/samples/context-api/Tabs.html
new file mode 100644
index 0000000000..8bbe89a017
--- /dev/null
+++ b/test/runtime/samples/context-api/Tabs.html
@@ -0,0 +1,48 @@
+
+
+<script context="module">
+	export const TABS = {};
+</script>
+
+<script>
+	import { setContext } from 'svelte';
+	import { writable } from 'svelte/store';
+
+	const tabs = [];
+	const panels = [];
+	const selected = writable(null);
+
+	setContext(TABS, {
+		registerTab: tab => {
+			tabs.push(tab);
+		},
+
+		unregisterTab: tab => {
+			const i = tabs.indexOf(tab);
+			tabs.splice(i, 1);
+		},
+
+		registerPanel: panel => {
+			panels.push(panel);
+
+			// if this is the first panel, select it
+			selected.update(current => current || panel);
+		},
+
+		unregisterPanel: panel => {
+			const i = panels.indexOf(panel);
+			panels.splice(i, 1);
+		},
+
+		selectTab: tab => {
+			const i = tabs.indexOf(tab);
+			selected.set(panels[i]);
+		},
+
+		selected
+	});
+</script>
+
+<div class="tabs">
+	<slot></slot>
+</div>
\ No newline at end of file
diff --git a/test/runtime/samples/context-api/_config.js b/test/runtime/samples/context-api/_config.js
new file mode 100644
index 0000000000..0a8c2f1780
--- /dev/null
+++ b/test/runtime/samples/context-api/_config.js
@@ -0,0 +1,73 @@
+export default {
+	html: `
+		<div class="tabs">
+			<div class="tab-list">
+				<button class="selected">small</button>
+				<button>large</button>
+			</div>
+
+			<h2>Small panel</h2>
+		</div>
+	`,
+
+	async test({ assert, component, target, window }) {
+		const click = new window.MouseEvent('click');
+		let buttons = target.querySelectorAll('button');
+
+		await buttons[1].dispatchEvent(click);
+
+		assert.htmlEqual(target.innerHTML, `
+			<div class="tabs">
+				<div class="tab-list">
+					<button>small</button>
+					<button class="selected">large</button>
+				</div>
+
+				<h2>Large panel</h2>
+			</div>
+		`);
+
+		component.show_medium = true;
+
+		assert.htmlEqual(target.innerHTML, `
+			<div class="tabs">
+				<div class="tab-list">
+					<button>small</button>
+					<button>medium</button>
+					<button class="selected">large</button>
+				</div>
+
+				<h2>Large panel</h2>
+			</div>
+		`);
+
+		buttons = target.querySelectorAll('button');
+
+		await buttons[1].dispatchEvent(click);
+
+		assert.htmlEqual(target.innerHTML, `
+			<div class="tabs">
+				<div class="tab-list">
+					<button>small</button>
+					<button class="selected">medium</button>
+					<button>large</button>
+				</div>
+
+				<h2>Medium panel</h2>
+			</div>
+		`);
+
+		component.show_medium = false;
+
+		assert.htmlEqual(target.innerHTML, `
+			<div class="tabs">
+				<div class="tab-list">
+					<button>small</button>
+					<button class="selected">large</button>
+				</div>
+
+				<h2>Large panel</h2>
+			</div>
+		`);
+	}
+};
\ No newline at end of file
diff --git a/test/runtime/samples/context-api/main.html b/test/runtime/samples/context-api/main.html
new file mode 100644
index 0000000000..ef3ee996d5
--- /dev/null
+++ b/test/runtime/samples/context-api/main.html
@@ -0,0 +1,30 @@
+<script>
+	import Tabs from './Tabs.html';
+	import TabList from './TabList.html';
+	import Tab from './Tab.html';
+	import TabPanel from './TabPanel.html';
+
+	export let show_medium = false;
+</script>
+
+<Tabs>
+	<TabList>
+		<Tab>small</Tab>
+		{#if show_medium}<Tab>medium</Tab>{/if}
+		<Tab>large</Tab>
+	</TabList>
+
+	<TabPanel>
+		<h2>Small panel</h2>
+	</TabPanel>
+
+	{#if show_medium}
+		<TabPanel>
+			<h2>Medium panel</h2>
+		</TabPanel>
+	{/if}
+
+	<TabPanel>
+		<h2>Large panel</h2>
+	</TabPanel>
+</Tabs>
\ No newline at end of file