diff --git a/examples/simple_trainer_2dgs.py b/examples/simple_trainer_2dgs.py
index 10900858..0ef299be 100644
--- a/examples/simple_trainer_2dgs.py
+++ b/examples/simple_trainer_2dgs.py
@@ -914,7 +914,7 @@ def render_traj(self, step: int):
 
             # write images
             canvas = torch.cat(
-                [colors, depths.repeat(1, 1, 3)], dim=1 if width > height else 1
+                [colors, depths.repeat(1, 1, 3)], dim=0 if width > height else 1
             )
             canvas = (canvas.cpu().numpy() * 255).astype(np.uint8)
             canvas_all.append(canvas)