diff --git a/examples/simple/tests/test.rs b/examples/simple/tests/test.rs
index eec67c330..f25102c16 100644
--- a/examples/simple/tests/test.rs
+++ b/examples/simple/tests/test.rs
@@ -3,14 +3,20 @@ use assert_cmd::prelude::*;
 use std::str;
 use anyhow::anyhow;
 use serde_json::{json, Value};
+
 #[test]
 fn test_integration() -> Result<(), anyhow::Error> {
 
+    Command::new("pulumi")
+        .args(["stack", "select", "test"])
+        .current_dir(".")
+        .assert()
+        .success();
+
     Command::new("pulumi")
         .args(["up", "-y"])
         .current_dir(".")
         .env("PULUMI_CONFIG_PASSPHRASE", " ")
-        .env("PULUMI_STACK", "test")
         .assert()
         .success();
 
@@ -18,7 +24,6 @@ fn test_integration() -> Result<(), anyhow::Error> {
         .args(["stack", "export"])
         .current_dir(".")
         .env("PULUMI_CONFIG_PASSPHRASE", " ")
-        .env("PULUMI_STACK", "test")
         .assert()
         .success();
     let stack = &binding