diff --git a/distorch_2.py b/distorch_2.py
index 9725ad0..e66c7fa 100644
--- a/distorch_2.py
+++ b/distorch_2.py
@@ -624,6 +624,8 @@ def override_class_with_distorch_safetensor_v2(cls):
             vram_string = ""
             if virtual_vram_gb > 0:
                 vram_string = f"{compute_device};{virtual_vram_gb};{donor_device}"
+            elif expert_mode_allocations:  # Only include compute device if there's an expert string
+                vram_string = compute_device
 
             full_allocation = f"{expert_mode_allocations}#{vram_string}" if expert_mode_allocations or vram_string else ""
             
diff --git a/pyproject.toml b/pyproject.toml
index 5740591..3e1008d 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.2.0"
+version = "2.2.1"
 license = {file = "LICENSE"}
 
 [project.urls]