diff --git a/scene/gaussian_model.py b/scene/gaussian_model.py
index 473887db89..909939322d 100644
--- a/scene/gaussian_model.py
+++ b/scene/gaussian_model.py
@@ -63,6 +63,7 @@ def __init__(self, sh_degree, optimizer_type="default"):
         self.optimizer = None
         self.percent_dense = 0
         self.spatial_lr_scale = 0
+        self.tmp_radii = None
         self.setup_functions()
 
     def capture(self):
@@ -361,7 +362,9 @@ def prune_points(self, mask):
 
         self.denom = self.denom[valid_points_mask]
         self.max_radii2D = self.max_radii2D[valid_points_mask]
-        self.tmp_radii = self.tmp_radii[valid_points_mask]
+        
+        if self.tmp_radii is not None:
+            self.tmp_radii = self.tmp_radii[valid_points_mask]
 
     def cat_tensors_to_optimizer(self, tensors_dict):
         optimizable_tensors = {}