diff --git a/mRemoteNGTests/Tree/ConnectionTreeTests.cs b/mRemoteNGTests/Tree/ConnectionTreeTests.cs
index ecf8818a5..be6f30204 100644
--- a/mRemoteNGTests/Tree/ConnectionTreeTests.cs
+++ b/mRemoteNGTests/Tree/ConnectionTreeTests.cs
@@ -17,7 +17,10 @@ namespace mRemoteNGTests.Tree
         public void Setup()
         {
             _connectionTreeModel = CreateConnectionTreeModel();
-            _connectionTree = new ConnectionTree();
+            _connectionTree = new ConnectionTree
+            {
+                PostSetupActions = new IConnectionTreeDelegate[] {new RootNodeExpander()}
+            };
         }
 
         [TearDown]