diff --git a/test/Microsoft.SqlTools.ServiceLayer.Test/LanguageServer/LanguageServiceTests.cs b/test/Microsoft.SqlTools.ServiceLayer.Test/LanguageServer/LanguageServiceTests.cs
index ddf82059..09706f6c 100644
--- a/test/Microsoft.SqlTools.ServiceLayer.Test/LanguageServer/LanguageServiceTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.Test/LanguageServer/LanguageServiceTests.cs
@@ -220,6 +220,7 @@ namespace Microsoft.SqlTools.ServiceLayer.Test.LanguageServices
             var autocompleteService = new AutoCompleteService();
             autocompleteService.ConnectionServiceInstance = connectionService;
             autocompleteService.InitializeService(Microsoft.SqlTools.ServiceLayer.Hosting.ServiceHost.Instance);
+            autocompleteService.ConnectionFactory = TestObjects.GetTestSqlConnectionFactory();
 
             // Open two connections
             ConnectParams connectionRequest1 = TestObjects.GetTestConnectionParams();