From 3f56275ff5675d32396cf84d4096ba5839cc7225 Mon Sep 17 00:00:00 2001
From: ComputerGuy <63362464+Ocean-OS@users.noreply.github.com>
Date: Thu, 28 Aug 2025 17:49:23 -0700
Subject: [PATCH] fix?

---
 .../phases/3-transform/client/transform-client.js   | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/packages/svelte/src/compiler/phases/3-transform/client/transform-client.js b/packages/svelte/src/compiler/phases/3-transform/client/transform-client.js
index e4b69300ee..111a9e601b 100644
--- a/packages/svelte/src/compiler/phases/3-transform/client/transform-client.js
+++ b/packages/svelte/src/compiler/phases/3-transform/client/transform-client.js
@@ -370,6 +370,10 @@ export function client_component(analysis, options) {
 			.../** @type {ESTree.Statement[]} */ (template.body)
 		]);
 
+		if (needs_store_cleanup) {
+			body.body.push(b.stmt(b.call('$$cleanup')));
+		}
+
 		component_block.body.push(b.stmt(b.call(`$.async_body`, b.arrow([], body, true))));
 	} else {
 		component_block.body.push(
@@ -452,11 +456,12 @@ export function client_component(analysis, options) {
 		component_block.body.push(to_push);
 	}
 
-	if (needs_store_cleanup) {
+	if (needs_store_cleanup && !analysis.instance.has_await) {
 		component_block.body.push(b.stmt(b.call('$$cleanup')));
-		if (component_returned_object.length > 0) {
-			component_block.body.push(b.return(b.id('$$pop')));
-		}
+	}
+
+	if (needs_store_cleanup && component_returned_object.length > 0) {
+		component_block.body.push(b.return(b.id('$$pop')));
 	}
 
 	if (analysis.uses_rest_props) {