diff --git a/test-cpu.sh b/test-cpu.sh
index a0f09a0..2cb3983 100755
--- a/test-cpu.sh
+++ b/test-cpu.sh
@@ -1,4 +1,4 @@
-version="pr-13"
+version="0.3.1"
 
 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/