diff --git a/hint_image_enchance.py b/hint_image_enchance.py
index e5a8112..cb2a069 100644
--- a/hint_image_enchance.py
+++ b/hint_image_enchance.py
@@ -216,7 +216,7 @@ def INPUT_TYPES(s):
     CATEGORY = "ControlNet Preprocessors"
 
     def execute(self, image):
-        _, _, H, W = image.shape
+        _, H, W, _ = image.shape
         return (W, H)
     
 NODE_CLASS_MAPPINGS = {