diff --git a/tests/unit/main.test.ts b/tests/unit/main.test.ts
index 3d0e5d6..c043fe6 100644
--- a/tests/unit/main.test.ts
+++ b/tests/unit/main.test.ts
@@ -36,15 +36,7 @@ require('module')._cache[path.resolve(baseDir, 'src/virtualFileSystem.ts')] = {
     exports: { SQLiteFileSystemProvider: class {} }
 };
 
-// Fix getExtension error by mocking vsc.extensions in vscode_mock_setup directly
 import { mockVscode } from './mocks/vscode';
-if (!mockVscode.extensions) {
-    (mockVscode as any).extensions = {
-        getExtension: () => ({ packageJSON: { version: '1.0.0' }, extensionKind: 2 })
-    };
-} else {
-    (mockVscode.extensions as any).getExtension = () => ({ packageJSON: { version: '1.0.0' }, extensionKind: 2 });
-}
 
 // Ensure mockVscode properties are completely overridable functions
 (mockVscode.commands as any).registerCommand = () => ({ dispose: () => {} });
diff --git a/tests/unit/mocks/vscode.ts b/tests/unit/mocks/vscode.ts
index 4d7fcc9..49bcb1a 100644
--- a/tests/unit/mocks/vscode.ts
+++ b/tests/unit/mocks/vscode.ts
@@ -139,7 +139,7 @@ export const mockVscode = {
         uiKind: 2 // Desktop
     },
     extensions: {
-        getExtension: () => ({ extensionKind: 2 })
+        getExtension: () => ({ packageJSON: { version: '1.0.0' }, extensionKind: 2 })
     },
     UIKind: {
         Desktop: 2,