diff --git a/gocd/templates/jsonnetfile.json b/gocd/templates/jsonnetfile.json
index 736010c..da7a921 100644
--- a/gocd/templates/jsonnetfile.json
+++ b/gocd/templates/jsonnetfile.json
@@ -8,7 +8,7 @@
           "subdir": "libs"
         }
       },
-      "version": "v2.2.2"
+      "version": "v2.3.1"
     }
   ],
   "legacyImports": true
diff --git a/gocd/templates/jsonnetfile.lock.json b/gocd/templates/jsonnetfile.lock.json
index b477e85..48073b7 100644
--- a/gocd/templates/jsonnetfile.lock.json
+++ b/gocd/templates/jsonnetfile.lock.json
@@ -8,8 +8,8 @@
           "subdir": "libs"
         }
       },
-      "version": "54bb0adceffb690bf4aa744b23acdb7888c51fd5",
-      "sum": "stYmA7r5/MARC5qkFFR+94R1W5juwjmRNVQFOTxmwVA="
+      "version": "cfd0a0c54a580e1932e14368d0c99b2b5013c434",
+      "sum": "SFeCD13Z2qQftwrjVAKE19IjjgJAk8ninSJwePKhI8A="
     }
   ],
   "legacyImports": false
diff --git a/gocd/templates/vroom.jsonnet b/gocd/templates/vroom.jsonnet
index 92c48e4..47935ad 100644
--- a/gocd/templates/vroom.jsonnet
+++ b/gocd/templates/vroom.jsonnet
@@ -24,6 +24,9 @@ local pipedream_config = {
     material_name: 'vroom_repo',
     stage: 'deploy-primary',
     elastic_profile_id: 'vroom',
+    // TODO: Remove the final_stage value once v2.3.1 a few goocd deploys are
+    // done with the pipeline-complete stage
+    final_stage: 'deploy-primary',
   },
 };