diff --git a/public/editormd/plugins/image-dialog/image-dialog.js b/public/editormd/plugins/image-dialog/image-dialog.js
index 6da63f1e5..708fd8881 100644
--- a/public/editormd/plugins/image-dialog/image-dialog.js
+++ b/public/editormd/plugins/image-dialog/image-dialog.js
@@ -113,13 +113,13 @@
                             }
 
                             this.hide().lockScreen(false).hideMask();
-
+                            cm.focus && cm.focus()
                             return false;
                         }],
 
                         cancel : [lang.buttons.cancel, function() {
                             this.hide().lockScreen(false).hideMask();
-
+                            cm.focus && cm.focus()
                             return false;
                         }]
                     }