diff --git a/packages/editor/src/extensions/image/component.tsx b/packages/editor/src/extensions/image/component.tsx
index becd762af..23ccdb0e3 100644
--- a/packages/editor/src/extensions/image/component.tsx
+++ b/packages/editor/src/extensions/image/component.tsx
@@ -105,6 +105,7 @@ export function ImageComponent(
   return (
     <>
       <Box
+        className=".image-container"
         sx={{
           ...getAlignmentStyles(node.attrs),
           height: float ? size.height : "unset",
diff --git a/packages/editor/styles/styles.css b/packages/editor/styles/styles.css
index f17819ee6..9fbd188b1 100644
--- a/packages/editor/styles/styles.css
+++ b/packages/editor/styles/styles.css
@@ -601,11 +601,11 @@ p > *::selection {
   pointer-events: none;
 }
 
-.ProseMirror > p[data-spacing="double"] .resizer {
+.ProseMirror > p[data-spacing="double"] .image-container {
   margin-bottom: 1em;
 }
 
-.ProseMirror > p[data-spacing="single"] .resizer {
+.ProseMirror > p[data-spacing="single"] .image-container {
   margin-top: 1em !important;
   margin-bottom: 1em;
 }