diff --git a/polyaxon/coherence_momentum/polyaxon-experiment-nomig.yml b/polyaxon/coherence_momentum/polyaxon-experiment-nomig.yml
index 0008b19..84a62a1 100644
--- a/polyaxon/coherence_momentum/polyaxon-experiment-nomig.yml
+++ b/polyaxon/coherence_momentum/polyaxon-experiment-nomig.yml
@@ -42,7 +42,7 @@ run:
       secret:
         secretName: "gcp-sa-credentials"
   container:
-    image: asia.gcr.io/nlp-hub/coherence-paradigm-refactored:0.5
+    image: asia.gcr.io/nlp-hub/coherence-paradigm-refactored:0.6
     imagePullPolicy: IfNotPresent
     workingDir: "{{ WORKING_DIR }}"
     command: ["/bin/bash","-c"]