diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index 2ff2123..385d43b 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -45,11 +45,12 @@ jobs:
     needs:
       - list-test-files
     container:
-      image: 10.0.13.31:5000/nvidia/cuda:128-ubuntu22.04_0822
+      image: 10.0.13.31:5000/nvidia/cuda:128-ubuntu22.04_0206
       volumes:
-        - /home/ci/models:/monster/data/model
-        - /home/ci/models/huggingface:/github/home/.cache/huggingface
-        - /home/ci/models/pyenv:/opt/pyenv
+        - /monster/ci/env/entrypoint.sh:/etc/profile.d/01-entrypoint.sh
+        - /monster/ci/models:/monster/data/model
+        - /monster/ci/huggingface:/github/home/.cache/huggingface
+        - /monster/ci/uv:/opt/uv
     strategy:
       fail-fast: false
       matrix:
@@ -63,25 +64,14 @@ jobs:
 
           echo "env name: $env_name"
 
-          if [ -d "$(pyenv root)/versions/$env_name" ]; then
-            echo "env exists, skip"
-            pyenv activate $env_name
-            pyenv local $env_name
-          else
-            echo "creating venv..."
-            pyenv virtualenv 3.12 "$env_name"
-            pyenv activate $env_name
-            pyenv local $env_name
-            bash -c "$(curl -L http://10.0.13.31/scripts/env/init_compiler_no_env.sh)" @ 128 2.8.0 3.12
-          fi
+          /opt/uv/setup_uv_venv.sh $env_name
 
       - name: install requirements
         run: |
-          pip install uv
-          uv pip install parameterized pytest pytest-xdist transformers -U
+          uv pip install parameterized pytest pytest-xdist jinja2 transformers sentencepiece torch tiktoken protobuf -U
 
       - name: install
-        run: pip install .
+        run: uv pip install .
 
       - name: test
         run: pytest -n 8 -v tests/${{ matrix.test_script }}
diff --git a/setup.py b/setup.py
index b314b98..a68d263 100644
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,7 @@
 
 from setuptools import find_packages, setup
 
-__version__ = "0.0.5"
+__version__ = "0.0.6"
 
 with open("requirements.txt") as f:
     requirements = f.read().splitlines()