diff --git a/examples/1_visualize_dataset.py b/examples/1_visualize_dataset.py
index a9406d1e..bfc2635f 100644
--- a/examples/1_visualize_dataset.py
+++ b/examples/1_visualize_dataset.py
@@ -17,4 +17,4 @@ video_paths = render_dataset(
     max_num_episodes=1,
 )
 print(video_paths)
-# ['outputs/visualize_dataset/example/episode_0.mp4']
+# ['outputs/visualize_dataset/example/episode_0_top.mp4']
diff --git a/tests/test_examples.py b/tests/test_examples.py
index 4263e452..17f6d600 100644
--- a/tests/test_examples.py
+++ b/tests/test_examples.py
@@ -15,7 +15,7 @@ def test_example_1():
         file_contents = file.read()
     exec(file_contents)
 
-    assert Path("outputs/visualize_dataset/example/episode_0.mp4").exists()
+    assert Path("outputs/visualize_dataset/example/episode_0_top.mp4").exists()
 
 
 def test_examples_3_and_2():