diff --git a/src/vs/workbench/api/browser/mainThreadWebview.ts b/src/vs/workbench/api/browser/mainThreadWebview.ts
index 78a62fbba9b..4c5b97e060d 100644
--- a/src/vs/workbench/api/browser/mainThreadWebview.ts
+++ b/src/vs/workbench/api/browser/mainThreadWebview.ts
@@ -285,6 +285,8 @@ export class MainThreadWebviews extends Disposable implements MainThreadWebviews
 					continue;
 				}
 
+				input.updateGroup(group.id);
+
 				const handle = this._webviewEditorInputs.getHandleForInput(input);
 				if (handle) {
 					viewStates[handle] = {