Backwards compatible fix for change in ComfyUi: https://github.com/comfyanonymous/ComfyUI/commit/4ba7fa0244badcf901f2b8ddbfb8539c6398672f
Backwards compatible fix for change in ComfyUi: https://github.com/comfyanonymous/ComfyUI/commit/4ba7fa0244badcf901f2b8ddbfb8539c6398672f