diff --git a/test/suite/commands.test.ts b/test/suite/commands.test.ts
index 317ca974..fbd7ed44 100644
--- a/test/suite/commands.test.ts
+++ b/test/suite/commands.test.ts
@@ -689,7 +689,7 @@ describe("dataSourceCommand2", () => {
       dummyFileContent.dataSource.selectedType = DataSourceTypes.API;
       getMetaStub.resolves(dummyMeta);
       getDataInsightsStub.resolves({ arrayBuffer: ab, error: "" });
-      isVisibleStub.returns(false);
+      ext.isResultsTabVisible = false;
       await dataSourceCommand.runDataSource(
         dummyFileContent as DataSourceFiles,
         insightsConn.connLabel,
@@ -708,7 +708,7 @@ describe("dataSourceCommand2", () => {
       dummyFileContent.dataSource.selectedType = DataSourceTypes.SQL;
       getMetaStub.resolves(dummyMeta);
       getDataInsightsStub.resolves({ arrayBuffer: ab, error: "" });
-      isVisibleStub.returns(false);
+      ext.isResultsTabVisible = false;
       await dataSourceCommand.runDataSource(
         dummyFileContent as DataSourceFiles,
         insightsConn.connLabel,