diff --git a/Tests/BasicTests/DownloadTests.cs b/Tests/BasicTests/DownloadTests.cs
index bba4da1..e26ec7f 100644
--- a/Tests/BasicTests/DownloadTests.cs
+++ b/Tests/BasicTests/DownloadTests.cs
@@ -24,8 +24,8 @@ namespace Tests.ParallelTests
 
         void ParallelDownload(int numberOfNodes, ByteSize filesize)
         {
-            var group = SetupCodexNodes(numberOfNodes).BringOnline();
-            var host = SetupCodexNodes(1).BringOnline()[0];
+            var group = SetupCodexNodes(numberOfNodes);
+            var host = SetupCodexNode();
 
             foreach (var node in group)
             {
diff --git a/Tests/BasicTests/UploadTests.cs b/Tests/BasicTests/UploadTests.cs
index a603524..5562170 100644
--- a/Tests/BasicTests/UploadTests.cs
+++ b/Tests/BasicTests/UploadTests.cs
@@ -23,8 +23,8 @@ namespace Tests.ParallelTests
         }
         void ParallelUpload(int numberOfNodes, ByteSize filesize)
         {
-            var group = SetupCodexNodes(numberOfNodes).BringOnline();
-            var host = SetupCodexNodes(1).BringOnline()[0];
+            var group = SetupCodexNodes(numberOfNodes);
+            var host = SetupCodexNode();
 
             foreach (var node in group)
             {