diff --git a/packages/api/src/mcp/registry/__tests__/MCPServersInitializer.test.ts b/packages/api/src/mcp/registry/__tests__/MCPServersInitializer.test.ts
index 30e898448f..c5eb1b7171 100644
--- a/packages/api/src/mcp/registry/__tests__/MCPServersInitializer.test.ts
+++ b/packages/api/src/mcp/registry/__tests__/MCPServersInitializer.test.ts
@@ -315,11 +315,11 @@ describe('MCPServersInitializer', () => {
       // First initialization - populates caches
       await MCPServersInitializer.initialize(testConfigs);
       expect(await registryStatusCache.isInitialized()).toBe(true);
-      expect(await registry.sharedAppServers.get('file_tools_server')).toBeDefined();
+      expect(await registry.getServerConfig('file_tools_server')).toBeDefined();
 
       // Simulate stale data: add an extra server that shouldn't be there
-      await registry.sharedAppServers.add('stale_server', testParsedConfigs.file_tools_server);
-      expect(await registry.sharedAppServers.get('stale_server')).toBeDefined();
+      await registry.addServer('stale_server', testConfigs.file_tools_server, 'CACHE');
+      expect(await registry.getServerConfig('stale_server')).toBeDefined();
 
       jest.clearAllMocks();
 
@@ -331,20 +331,20 @@ describe('MCPServersInitializer', () => {
       await MCPServersInitializer.initialize(testConfigs);
 
       // Verify stale server was removed (cache was reset)
-      expect(await registry.sharedAppServers.get('stale_server')).toBeUndefined();
+      expect(await registry.getServerConfig('stale_server')).toBeUndefined();
 
       // Verify new servers are present
-      expect(await registry.sharedAppServers.get('file_tools_server')).toBeDefined();
-      expect(await registry.sharedUserServers.get('oauth_server')).toBeDefined();
+      expect(await registry.getServerConfig('file_tools_server')).toBeDefined();
+      expect(await registry.getServerConfig('oauth_server')).toBeDefined();
 
       // Verify inspector was called again (re-initialization happened)
       expect(mockInspect).toHaveBeenCalled();
     });
 
     it('should not re-initialize on subsequent calls within same process', async () => {
-      // First initialization
+      // First initialization (5 servers in testConfigs)
       await MCPServersInitializer.initialize(testConfigs);
-      expect(mockInspect).toHaveBeenCalledTimes(4);
+      expect(mockInspect).toHaveBeenCalledTimes(5);
 
       jest.clearAllMocks();