diff --git a/octopusdeploy_framework/testing_container_test.go b/octopusdeploy_framework/testing_container_test.go
index 7aa3fb00..6b8c0f90 100644
--- a/octopusdeploy_framework/testing_container_test.go
+++ b/octopusdeploy_framework/testing_container_test.go
@@ -29,14 +29,14 @@ func TestMain(m *testing.M) {
 	if *createSharedContainer {
 
 		testFramework := test.OctopusContainerTest{}
-		octoContainer, octoClient, sqlServerContainer, network, err = testFramework.ArrangeContainer()
+		octoContainer, _, sqlServerContainer, network, err = testFramework.ArrangeContainer()
 		if err != nil {
 			log.Printf("Failed to arrange containers: (%s)", err.Error())
 		}
 
 		os.Setenv("OCTOPUS_URL", octoContainer.URI)
 		os.Setenv("OCTOPUS_APIKEY", test.ApiKey)
-
+		octoClient, err = octoclient.CreateClient(octoContainer.URI, "", test.ApiKey)
 		code := m.Run()
 		ctx := context.Background()