diff --git a/src/paper/generate_attention_figure.py b/src/paper/generate_attention_figure.py
index 36cea7a..97674c4 100644
--- a/src/paper/generate_attention_figure.py
+++ b/src/paper/generate_attention_figure.py
@@ -79,7 +79,6 @@ def generate_attention_figure(cfg_dict):
         step_tracker=None,
     )
     model_wrapper.eval()
-    dataset = iter(get_dataset(cfg.dataset, "test", None))
 
     # Create a dataset that always returns the desired scene.
     view_sampler_cfg = ViewSamplerArbitraryCfg(