diff --git a/examples/collectors/isaaclab_rnn_ppo_memory.py b/examples/collectors/isaaclab_rnn_ppo_memory.py
index 49dd5ac9484..d9f6ad0a3f6 100644
--- a/examples/collectors/isaaclab_rnn_ppo_memory.py
+++ b/examples/collectors/isaaclab_rnn_ppo_memory.py
@@ -253,6 +253,12 @@ def parse_args() -> argparse.Namespace:
             "'compact' for the compile-friendly compact shifted path."
         ),
     )
+    parser.add_argument(
+        "--gae-compact-cat-dim",
+        choices=["batch", "time"],
+        default="batch",
+        help="Concatenation dimension used by GAE shifted='compact'.",
+    )
     parser.add_argument(
         "--deactivate-vmap",
         action=argparse.BooleanOptionalAction,
@@ -342,6 +348,7 @@ def main() -> None:
         value_network=full_value,
         average_gae=False,
         shifted=gae_shifted,
+        compact_cat_dim=args.gae_compact_cat_dim,
         deactivate_vmap=args.deactivate_vmap,
         device=train_device,
     )