diff --git a/projects/nerf/nerf/eval_video_utils.py b/projects/nerf/nerf/eval_video_utils.py
index b35dde73..0720624b 100644
--- a/projects/nerf/nerf/eval_video_utils.py
+++ b/projects/nerf/nerf/eval_video_utils.py
@@ -9,6 +9,7 @@ from typing import Tuple
 
 import torch
 from pytorch3d.renderer import PerspectiveCameras, look_at_view_transform
+from torch.utils.data.dataset import Dataset
 
 
 def generate_eval_video_cameras(
@@ -18,7 +19,7 @@ def generate_eval_video_cameras(
     trajectory_scale: float = 0.2,
     scene_center: Tuple[float, float, float] = (0.0, 0.0, 0.0),
     up: Tuple[float, float, float] = (0.0, 0.0, 1.0),
-) -> dict:
+) -> Dataset[torch.Tensor]:
     """
     Generate a camera trajectory for visualizing a NeRF model.