diff --git a/src/services/marketplace/__tests__/marketplace-setting-check.test.ts b/src/services/marketplace/__tests__/marketplace-setting-check.test.ts
index 30b8530c3c..1d86939509 100644
--- a/src/services/marketplace/__tests__/marketplace-setting-check.test.ts
+++ b/src/services/marketplace/__tests__/marketplace-setting-check.test.ts
@@ -18,6 +18,7 @@ jest.mock("vscode", () => ({
 const mockProvider = {
 	getState: jest.fn(),
 	postStateToWebview: jest.fn(),
+	getStateToPostToWebview: jest.fn(),
 } as any
 
 const mockMarketplaceManager = {
@@ -34,6 +35,9 @@ describe("Marketplace Setting Check", () => {
 		mockProvider.getState.mockResolvedValue({
 			experiments: { marketplace: false },
 		})
+		mockProvider.getStateToPostToWebview.mockResolvedValue({
+			marketplaceEnabled: false,
+		})
 
 		const message = {
 			type: "filterMarketplaceItems" as const,
@@ -52,6 +56,9 @@ describe("Marketplace Setting Check", () => {
 		mockProvider.getState.mockResolvedValue({
 			experiments: { marketplace: true },
 		})
+		mockProvider.getStateToPostToWebview.mockResolvedValue({
+			marketplaceEnabled: true,
+		})
 
 		const message = {
 			type: "filterMarketplaceItems" as const,
@@ -74,6 +81,9 @@ describe("Marketplace Setting Check", () => {
 		mockProvider.getState.mockResolvedValue({
 			experiments: { marketplace: false },
 		})
+		mockProvider.getStateToPostToWebview.mockResolvedValue({
+			marketplaceEnabled: false,
+		})
 
 		const mockInstallMarketplaceItem = jest.fn()
 		const mockMarketplaceManagerWithInstall = {