diff --git a/custom_nodes/annotator/__init__.py b/custom_nodes/annotator/__init__.py
index ecb01650b..730e7507f 100644
--- a/custom_nodes/annotator/__init__.py
+++ b/custom_nodes/annotator/__init__.py
@@ -16,11 +16,8 @@ def common_annotator_call(annotator_callback, tensor_image, *args):
     call_result = annotator_callback(img_tensor_to_np(tensor_image), *args)
     if type(annotator_callback) is openpose.OpenposeDetector:
         return (HWC3(call_result[0]),call_result[1])
-    # if type(call_result) is tuple:
-    #     for i in range(len(call_result)):
-    #         call_result[i] = HWC3(call_result[i])
-    # else:
-    #     call_result = HWC3(call_result)
+    if type(annotator_callback) is midas.MidasDetector():
+        return (HWC3(call_result[0]),HWC3(call_result[1]))
     return HWC3(call_result)