diff --git a/distorch_2.py b/distorch_2.py
index 1922875..fea6181 100644
--- a/distorch_2.py
+++ b/distorch_2.py
@@ -301,6 +301,8 @@ def register_patched_safetensor_modelpatcher():
 
             self.model.current_weight_patches_uuid = self.patches_uuid
 
+            self.model.device = device_to
+
             logger.info("[MultiGPU DisTorch V2] DisTorch loading completed.")
             logger.info(f"[MultiGPU DisTorch V2] Total memory: {mem_counter / (1024 * 1024):.2f}MB")
 
diff --git a/pyproject.toml b/pyproject.toml
index 8fcc17e..27c65bb 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-multigpu"
 description = "Provides a suite of custom nodes to manage multiple GPUs for ComfyUI, including advanced model offloading for both GGUF and Safetensor formats with DisTorch, and bespoke MultiGPU support for WanVideoWrapper and other custom nodes."
-version = "2.5.9"
+version = "2.5.10"
 license = {file = "LICENSE"}
 
 [project.urls]