diff --git a/src/sql/workbench/api/common/sqlExtHost.api.impl.ts b/src/sql/workbench/api/common/sqlExtHost.api.impl.ts
index 2fec9b3d7c..9e5d747643 100644
--- a/src/sql/workbench/api/common/sqlExtHost.api.impl.ts
+++ b/src/sql/workbench/api/common/sqlExtHost.api.impl.ts
@@ -474,8 +474,8 @@ export function createAdsApiFactory(accessor: ServicesAccessor): IAdsExtensionAp
 					return extHostQueryEditor.$connect(fileUri, connectionId);
 				},
 
-				runQuery(fileUri: string, options?: Map<string, string>): void {
-					extHostQueryEditor.$runQuery(fileUri);
+				runQuery(fileUri: string, options?: Map<string, string>, runCurrentQuery: boolean = true): void {
+					extHostQueryEditor.$runQuery(fileUri, runCurrentQuery);
 				},
 
 				registerQueryEventListener(listener: azdata.queryeditor.QueryEventListener): void {