diff --git a/genai/image-generation/imggen_virtual-try-on-with-txt-img.js b/genai/image-generation/imggen_virtual-try-on-with-txt-img.js
index ff0686357a..2a4754b5ea 100644
--- a/genai/image-generation/imggen_virtual-try-on-with-txt-img.js
+++ b/genai/image-generation/imggen_virtual-try-on-with-txt-img.js
@@ -48,7 +48,7 @@ async function virtualTryOn(
   };
 
   const image = await client.models.recontextImage({
-    model: 'virtual-try-on-preview-08-04',
+    model: 'virtual-try-on-001',
     source: source,
   });
 
diff --git a/genai/image-generation/test/imggen-util.js b/genai/image-generation/test/imggen-util.js
index 36e77bb2d8..733266bb81 100644
--- a/genai/image-generation/test/imggen-util.js
+++ b/genai/image-generation/test/imggen-util.js
@@ -15,7 +15,8 @@
 const {Storage} = require('@google-cloud/storage');
 const {format} = require('date-fns');
 
-const gcsOutputBucket = 'nodejs-docs-samples-tests';
+const gcsOutputBucket =
+  process.env.GCLOUD_STORAGE_BUCKET || 'nodejs-docs-samples-tests';
 
 module.exports.createOutputGcsUri = async function () {
   const prefix = `text_output/${format(new Date(), 'yyyy-MM-dd-HH-mm-ss')}`;