diff --git a/tensorrt_llm/_torch/speculative/dspark.py b/tensorrt_llm/_torch/speculative/dspark.py
index f2597f5e9ef3..80425833b0be 100644
--- a/tensorrt_llm/_torch/speculative/dspark.py
+++ b/tensorrt_llm/_torch/speculative/dspark.py
@@ -404,7 +404,7 @@ def _draft_gen_block_batched(
         )
         return block_logits
 
-    def forward(
+    def _forward_impl(
         self,
         input_ids,
         position_ids,