diff --git a/vllm_npu/worker/worker_v1.py b/vllm_npu/worker/worker_v1.py
index 82810e3..5e33b0d 100644
--- a/vllm_npu/worker/worker_v1.py
+++ b/vllm_npu/worker/worker_v1.py
@@ -85,11 +85,11 @@ class NPUWorker(WorkerBase):
 
         # Initialize distributed (HCCL)
         init_distributed_environment(
-            self.vllm_config,
-            self.rank,
-            self.distributed_init_method,
-            self.local_rank,
-            "hccl",
+            world_size=self.parallel_config.world_size,
+            rank=self.rank,
+            distributed_init_method=self.distributed_init_method,
+            local_rank=self.local_rank,
+            backend="hccl",
         )
 
         # Set random seed