diff --git a/nerfactory/data/structs.py b/nerfactory/data/structs.py
index 089772b49..437d656cd 100644
--- a/nerfactory/data/structs.py
+++ b/nerfactory/data/structs.py
@@ -22,7 +22,7 @@ from typing import List, Optional
 import torch
 from torchtyping import TensorType
 
-from pyrad.cameras.rays import RayBundle
+from nerfactory.cameras.rays import RayBundle
 
 
 @dataclass