diff --git a/.github/workflows/unit_test_200gb_CI.yml b/.github/workflows/unit_test_200gb_CI.yml
index 0344a1378..43ea9b275 100644
--- a/.github/workflows/unit_test_200gb_CI.yml
+++ b/.github/workflows/unit_test_200gb_CI.yml
@@ -37,7 +37,7 @@ jobs:
           mode: start
           github-token: ${{ secrets.GH_PERSONAL_ACCESS_TOKEN }}
           ec2-image-id:  ${{ secrets.AMD_200GB_EC2_IMAGE_ID }}
-          ec2-instance-type: t3.xlarge
+          ec2-instance-type: t3.2xlarge
           subnet-id: ${{ secrets.AMD_SUBNET_ID }}
           security-group-id: ${{ secrets.AMD_SECURITY_GROUP_ID }}
 
diff --git a/tests/core/index_management/test_index_management.py b/tests/core/index_management/test_index_management.py
index 373d16d3f..49319f469 100644
--- a/tests/core/index_management/test_index_management.py
+++ b/tests/core/index_management/test_index_management.py
@@ -17,7 +17,7 @@
 class TestIndexManagement(MarqoTestCase):
 
     def setUp(self):
-        self.index_management = IndexManagement(self.vespa_client)
+        self.index_management = IndexManagement(self.vespa_client, self.zookeeper_client)
 
     def test_bootstrap_vespa_doesNotExist_successful(self):
         settings_schema_name = 'a' + str(uuid.uuid4()).replace('-', '')