diff --git a/pytorch3d/implicitron/models/view_pooler/feature_aggregator.py b/pytorch3d/implicitron/models/view_pooler/feature_aggregator.py
index 442466bf..b51cd0c0 100644
--- a/pytorch3d/implicitron/models/view_pooler/feature_aggregator.py
+++ b/pytorch3d/implicitron/models/view_pooler/feature_aggregator.py
@@ -229,7 +229,7 @@ class ReductionFeatureAggregator(torch.nn.Module, FeatureAggregatorBase):
             masks_sampled.device,
             self.exclude_target_view,
         )
-        aggr_weigths = masks_sampled * sampling_mask
+        aggr_weigths = masks_sampled[..., 0] * sampling_mask[..., None]
         feats_aggregated = {
             k: _avgmaxstd_reduction_function(
                 f,