diff --git a/.github/workflows/unit_test_4gpu.yaml b/.github/workflows/unit_test_4gpu.yaml
index b71e545d..fa932df0 100644
--- a/.github/workflows/unit_test_4gpu.yaml
+++ b/.github/workflows/unit_test_4gpu.yaml
@@ -20,15 +20,16 @@ jobs:
       docker-image: "nvidia/cuda:12.4.1-runtime-ubuntu22.04"
       repository: "pytorch/torchtitan"
       upload-artifact: "outputs"
-        # conda create -n "test" python=3.10
-        # conda activate test
+        #         ~/miniconda3/bin/conda init bash
       script: |
         apt update && apt install -y wget
         mkdir -p ~/miniconda3
         wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
         bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
         rm -rf ~/miniconda3/miniconda.sh
-        ~/miniconda3/bin/conda init bash
+        source ~miniconda3/bin/activate
+        conda create -n "test" python=3.10
+        conda activate test
         conda install -y -q git clang clangxx
         export CC=clang
         export CXX=clangxx