diff --git a/nerfstudio/data/datamanagers/random_cameras_datamanager.py b/nerfstudio/data/datamanagers/random_cameras_datamanager.py
index a5868f54c..391e6a503 100644
--- a/nerfstudio/data/datamanagers/random_cameras_datamanager.py
+++ b/nerfstudio/data/datamanagers/random_cameras_datamanager.py
@@ -25,6 +25,7 @@ from typing import Dict, List, Tuple, Type, Union
 import torch
 from rich.progress import Console
 from torch.nn import Parameter
+from torch import Tensor
 from typing_extensions import Literal
 
 from nerfstudio.cameras.cameras import Cameras