diff --git a/README.md b/README.md
index dcdaa353c2db..9b5f301c99b0 100644
--- a/README.md
+++ b/README.md
@@ -110,7 +110,6 @@ ComfyUI follows a weekly release cycle every Friday, with three interconnected r
 
 2. **[ComfyUI Desktop](https://github.com/Comfy-Org/desktop)**
    - Builds a new release using the latest stable core version
-   - Version numbers match the core release (e.g., Desktop v1.7.0 uses Core v1.7.0)
 
 3. **[ComfyUI Frontend](https://github.com/Comfy-Org/ComfyUI_frontend)**
    - Weekly frontend updates are merged into the core repository
diff --git a/nodes.py b/nodes.py
index 0a9db139351c..95e831b8b437 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1940,7 +1940,7 @@ def expand_image(self, image, left, top, right, bottom, feathering):
 
         mask[top:top + d2, left:left + d3] = t
 
-        return (new_image, mask)
+        return (new_image, mask.unsqueeze(0))
 
 
 NODE_CLASS_MAPPINGS = {