diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py
index fa617d5d46..231dbe7765 100644
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@@ -122,7 +122,8 @@ class Unsloth{RLConfig_name}({RLConfig_name}):
         metadata = {{'help': 'Chunk size to reduce memory usage'}},
     )
     def __init__({RLConfig_arguments},
-        vllm_sampling_params = None,
+        vllm_sampling_params = vllm_sampling_params,
+        unsloth_num_chunks = unsloth_num_chunks,
         **kwargs,
     ):
 {RLConfig_extra_args}