diff --git a/kilosort4/Dockerfile b/kilosort4/Dockerfile
index e139b3a..3647f63 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.16
+RUN pip install kilosort==4.0.18
diff --git a/kilosort4/build.sh b/kilosort4/build.sh
index 8f7e3ca..32763ba 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.16_cuda-12.0.0 .
+docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0.18_cuda-12.0.0 .