diff --git a/tests/nerf/test_sampler.py b/tests/nerf/test_sampler.py
index 8e5ce7711..951b07c98 100644
--- a/tests/nerf/test_sampler.py
+++ b/tests/nerf/test_sampler.py
@@ -2,7 +2,6 @@
 Test samplers
 """
 import torch
-from mattport.nerf.field_modules.field_heads import FieldHeadNames
 from mattport.nerf.sampler import PDFSampler, UniformSampler
 
 from mattport.structures.rays import RayBundle