diff --git a/ui/media/js/dnd.js b/ui/media/js/dnd.js
index 0b22c4c9..e3b8cc9f 100644
--- a/ui/media/js/dnd.js
+++ b/ui/media/js/dnd.js
@@ -357,6 +357,7 @@ function restoreTaskToUI(task, fieldsToSkip) {
         initImagePreview.addEventListener('load', function() {
             if (Boolean(task.reqBody.mask)) {
                 imageInpainter.setImg(task.reqBody.mask)
+                maskSetting.checked = true
             }
         }, { once: true })
         initImagePreview.src = task.reqBody.init_image