diff --git a/python/jnerf/models/position_encoders/hash_encoder/hash_encoder.py b/python/jnerf/models/position_encoders/hash_encoder/hash_encoder.py
index ed20f989..9c904b67 100644
--- a/python/jnerf/models/position_encoders/hash_encoder/hash_encoder.py
+++ b/python/jnerf/models/position_encoders/hash_encoder/hash_encoder.py
@@ -10,12 +10,13 @@ def __init__(self, n_pos_dims=3, n_features_per_level=2, n_levels=16, base_resol
         self.cfg = get_cfg()
         using_fp16 = self.cfg.fp16
         aabb_scale = self.cfg.dataset_obj.aabb_scale
+        n_rays_per_batch = self.cfg.n_rays_per_batch
         self.hash_func = self.cfg.hash_func
         self.hash_func_header = f"""
 #define get_index(p0,p1,p2) {self.hash_func}
         """
         self.encoder = GridEncode(self.hash_func_header, aabb_scale=aabb_scale, n_pos_dims=3, n_features_per_level=2,
-                                  n_levels=16, base_resolution=16, log2_hashmap_size=19, using_fp16=using_fp16)
+                                  n_levels=16, base_resolution=16, log2_hashmap_size=19, using_fp16=using_fp16, n_rays_per_batch=n_rays_per_batch)
         self.grad_type = 'float32'
         if using_fp16:
             self.grad_type = 'float16'