diff --git a/main.nf b/main.nf
index e442b72..c017c6f 100755
--- a/main.nf
+++ b/main.nf
@@ -8,7 +8,7 @@ params.msa_server = 'https://api.colabfold.com'
 params.num_samples = 5
 
 process CHAI1 {
-    container 'harbor.cluster.omic.ai/omic/chai1:latest'
+    container 'harbor.cluster.omic.ai/omic/chai1:v2'
     publishDir params.outdir, mode: 'copy'
     stageInMode 'copy'
     maxForks 1
diff --git a/nextflow.config b/nextflow.config
index c62638d..65943e2 100755
--- a/nextflow.config
+++ b/nextflow.config
@@ -3,7 +3,7 @@ docker {
 }
 
 process {
-    container = 'harbor.cluster.omic.ai/omic/chai1:latest'
+    container = 'harbor.cluster.omic.ai/omic/chai1:v2'
     memory = '32 GB'
     cpus = 4
 }