diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
index 7c0ab6c..769ea65 100644
--- a/.github/workflows/docker-publish.yml
+++ b/.github/workflows/docker-publish.yml
@@ -107,6 +107,22 @@ jobs:
       - name: Set up QEMU
         uses: docker/setup-qemu-action@v2
 
+      - name: Free Disk Space (Ubuntu)
+        uses: jlumbroso/free-disk-space@main
+        with:
+          # this might remove tools that are actually needed,
+          # if set to "true" but frees about 6 GB
+          tool-cache: false
+
+          # all of these default to true, but feel free to set to
+          # "false" if necessary for your workflow
+          android: true
+          dotnet: true
+          haskell: true
+          large-packages: true
+          docker-images: true
+          swap-storage: true
+
       # Build and push Docker image with Buildx
       # https://github.com/docker/build-push-action
       - name: Build and push Docker image (JAX conda CPU)
@@ -121,6 +137,23 @@ jobs:
           build-args: |
             JAX_VERSION=${{ env.JAX_VERSION }}
 
+      - name: Free Disk Space (Ubuntu)
+        uses: jlumbroso/free-disk-space@main
+        with:
+          # this might remove tools that are actually needed,
+          # if set to "true" but frees about 6 GB
+          tool-cache: false
+
+          # all of these default to true, but feel free to set to
+          # "false" if necessary for your workflow
+          android: true
+          dotnet: true
+          haskell: true
+          large-packages: true
+          docker-images: true
+          swap-storage: true
+
+
       - name: Build and push Docker image (JAX conda GPU)
         id: build-and-push-jax-conda-gpu
         uses: docker/build-push-action@v5
diff --git a/test-cpu.sh b/test-cpu.sh
index 2cb3983..da05c6f 100755
--- a/test-cpu.sh
+++ b/test-cpu.sh
@@ -1,6 +1,6 @@
-version="0.3.1"
+version="0.3.5"
 
 docker pull ghcr.io/yfukai/basicpy-docker-mcmicro:${version}
-docker run --rm -v $(pwd)/testdata:/data ghcr.io/yfukai/basicpy-docker-mcmicro:${version} -d=cpu -i /data/exemplar-001-cycle-06.ome.tiff -o /data/
+docker run --rm -v $(pwd)/testdata:/data ghcr.io/yfukai/basicpy-docker-mcmicro:${version} /opt/main.py -d=cpu -i /data/exemplar-001-cycle-06.ome.tiff -o /data/
 docker pull ghcr.io/yfukai/basicpy-docker-mcmicro:latest
-docker run --rm -v $(pwd)/testdata:/data ghcr.io/yfukai/basicpy-docker-mcmicro:latest -d=cpu -i /data/exemplar-001-cycle-06.ome.tiff -o /data/
+docker run --rm -v $(pwd)/testdata:/data ghcr.io/yfukai/basicpy-docker-mcmicro:latest /opt/main.py -d=cpu -i /data/exemplar-001-cycle-06.ome.tiff -o /data/
diff --git a/test-gpu.sh b/test-gpu.sh
index 0029744..4c58d65 100755
--- a/test-gpu.sh
+++ b/test-gpu.sh
@@ -1,4 +1,4 @@
-version="0.3.1"
+version="0.3.5"
 
 docker pull ghcr.io/yfukai/basicpy-docker-mcmicro:${version}-cuda
 docker run --rm -v $(pwd)/testdata:/data ghcr.io/yfukai/basicpy-docker-mcmicro:${version}-cuda -d=gpu -i /data/exemplar-001-cycle-06.ome.tiff -o /data/