diff --git a/comfyui-hydit/nodes.py b/comfyui-hydit/nodes.py
index e228982..034cd59 100644
--- a/comfyui-hydit/nodes.py
+++ b/comfyui-hydit/nodes.py
@@ -359,7 +359,7 @@ def sample(self, maked_pipeline, positive, negative, batch_size, width, height,
             img = 255. * image.cpu().numpy()
             # print(i.size())
             img = Image.fromarray(np.clip(img[0], 0, 255).astype(np.uint8))
-            img = img.convert('RGB').resize((height, width))
+            img = img.convert('RGB').resize((width, height))
             image = norm_transform(img)
             image = image.unsqueeze(0).cuda()
             # print(image.size)