diff --git a/examples/radiance_fields/ngp.py b/examples/radiance_fields/ngp.py
index 44b81c5..3a889ea 100644
--- a/examples/radiance_fields/ngp.py
+++ b/examples/radiance_fields/ngp.py
@@ -212,8 +212,8 @@ def forward(
             assert (
                 positions.shape == directions.shape
             ), f"{positions.shape} v.s. {directions.shape}"
-            density, embedding = self.query_density(positions, return_feat=True)
-            rgb = self._query_rgb(directions, embedding=embedding)
+        density, embedding = self.query_density(positions, return_feat=True)
+        rgb = self._query_rgb(directions, embedding=embedding)
         return rgb, density  # type: ignore