diff --git a/test/test_artifacts/v1/scripts/run_autogluon_tests.sh b/test/test_artifacts/v1/scripts/run_autogluon_tests.sh
index 036d1432..9a669143 100644
--- a/test/test_artifacts/v1/scripts/run_autogluon_tests.sh
+++ b/test/test_artifacts/v1/scripts/run_autogluon_tests.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-AUTOGLUON_VERSION=$(micromamba list | grep autogluon | tr -s ' ' | cut -d ' ' -f 3)
+AUTOGLUON_VERSION=$(micromamba list | grep autogluon | tr -s ' ' | cut -d ' ' -f 3 | head -n 1)
 git checkout tags/v$AUTOGLUON_VERSION
 
 # Run autogluon quick start as end-to-end check
diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py
index bbaa4575..b3d910f3 100644
--- a/test/test_dockerfile_based_harness.py
+++ b/test/test_dockerfile_based_harness.py
@@ -126,7 +126,8 @@ def _validate_docker_images(dockerfile_path: str, required_packages: List[str],
         image, _ = _docker_client.images.build(path=test_artifacts_path,
                                                dockerfile=dockerfile_path,
                                                tag=dockerfile_path.lower().replace('.', '-'),
-                                               rm=True, buildargs={'COSMOS_IMAGE': docker_image_identifier})
+                                               rm=True, buildargs={'COSMOS_IMAGE': docker_image_identifier},
+                                               shm_size='256m')
     except BuildError as e:
         for line in e.build_log:
             if 'stream' in line: