diff --git a/packages/editor/src/extensions/image/component.tsx b/packages/editor/src/extensions/image/component.tsx
index f323b9073..5371c597b 100644
--- a/packages/editor/src/extensions/image/component.tsx
+++ b/packages/editor/src/extensions/image/component.tsx
@@ -100,6 +100,7 @@ export function ImageComponent(
       <Box
         sx={{
           ...getAlignmentStyles(node.attrs),
+          height: float ? height : "unset",
           position: "relative",
           mt: isSVG ? `24px` : 0,
           ":hover .drag-handle, :active .drag-handle": {
@@ -220,8 +221,8 @@ export function ImageComponent(
                 position: "absolute",
                 top: 0,
                 left: 0,
-                width: editor.isEditable ? "100%" : width,
-                height: editor.isEditable ? "100%" : height,
+                width: editor.isEditable && !float ? "100%" : width,
+                height: editor.isEditable && !float ? "100%" : height,
                 bg: "background-secondary",
                 border: selected
                   ? "2px solid var(--accent)"