diff --git a/src/sql/parts/query/editor/gridPanel.ts b/src/sql/parts/query/editor/gridPanel.ts
index 5ca2fbbc77..d2cc6302a1 100644
--- a/src/sql/parts/query/editor/gridPanel.ts
+++ b/src/sql/parts/query/editor/gridPanel.ts
@@ -592,6 +592,8 @@ class GridTable<T> extends Disposable implements IView {
 
 		this.restoreScrollState();
 
+		this.rebuildActionBar();
+
 		// Setting the active cell resets the selection so save it here
 		let savedSelection = this.state.selection;