diff --git a/src/vs/workbench/contrib/inlineChat/browser/inlineChatStrategies.ts b/src/vs/workbench/contrib/inlineChat/browser/inlineChatStrategies.ts
index 28e5b29c7a9..01df414ec2a 100644
--- a/src/vs/workbench/contrib/inlineChat/browser/inlineChatStrategies.ts
+++ b/src/vs/workbench/contrib/inlineChat/browser/inlineChatStrategies.ts
@@ -582,9 +582,9 @@ export class LiveStrategy extends EditModeStrategy {
 
 								data.lensActionsViewZoneIds?.forEach(viewZoneAccessor.removeZone);
 								data.lensActionsViewZoneIds = undefined;
-								lensActions?.dispose();
 							});
 
+							lensActions?.dispose();
 							overlay?.dispose();
 						};