diff --git a/cosmos1/models/autoregressive/nemo/post_training/README.md b/cosmos1/models/autoregressive/nemo/post_training/README.md
index ef0f7e1..8ecedb3 100644
--- a/cosmos1/models/autoregressive/nemo/post_training/README.md
+++ b/cosmos1/models/autoregressive/nemo/post_training/README.md
@@ -147,7 +147,7 @@ Complete the following steps to post-train the Cosmos-1.0-Autoregressive-4B mode
 
    # Optionally, you can monitor training progress with Weights and Biases (wandb).
    export WANDB_API_KEY="</your/wandb/api/key>"
-   export WANDB_PROJECT_NAME="cosmos-autoregressive-nemo-finetuning"
+   export WANDB_PROJECT="cosmos-autoregressive-nemo-finetuning"
    export WANDB_RUN_ID="cosmos_autoregressive_4b_finetune"
    ```
 2. Run the following command for Cosmos-1.0-Autoregressive-4B post-training:
diff --git a/cosmos1/models/diffusion/nemo/post_training/README.md b/cosmos1/models/diffusion/nemo/post_training/README.md
index ba8c3de..0778546 100644
--- a/cosmos1/models/diffusion/nemo/post_training/README.md
+++ b/cosmos1/models/diffusion/nemo/post_training/README.md
@@ -150,7 +150,7 @@ Complete the following steps to post-train the Cosmos-1.0-Diffusion-7B-Text2Worl
 
    # Optionally, you can monitor training progress with Weights and Biases (wandb).
    export WANDB_API_KEY="</your/wandb/api/key>"
-   export WANDB_PROJECT_NAME="cosmos-diffusion-nemo-post-training"
+   export WANDB_PROJECT="cosmos-diffusion-nemo-post-training"
    export WANDB_RUN_ID="cosmos_diffusion_7b_text2world_finetune"
    ```
 2. Run the following command for Cosmos-Diffusion-Text2World-7B general post-training: