diff --git a/kilosort4/Dockerfile b/kilosort4/Dockerfile
index 001da23..5f64372 100644
--- a/kilosort4/Dockerfile
+++ b/kilosort4/Dockerfile
@@ -35,4 +35,4 @@ RUN conda init bash
 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
 # install kilosort4
-RUN pip install kilosort==4.0
+RUN pip install kilosort==4.0.6
diff --git a/kilosort4/build.sh b/kilosort4/build.sh
index 1c735c3..ebb5cae 100755
--- a/kilosort4/build.sh
+++ b/kilosort4/build.sh
@@ -1,3 +1,3 @@
 #!/bin/bash
 
-docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0_cuda-12.0.0 .
+docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0.6_cuda-12.0.0 .