diff --git a/services/web/frontend/js/features/source-editor/extensions/visual/atomic-decorations.ts b/services/web/frontend/js/features/source-editor/extensions/visual/atomic-decorations.ts
index 1028daeff9..95b9d2f052 100644
--- a/services/web/frontend/js/features/source-editor/extensions/visual/atomic-decorations.ts
+++ b/services/web/frontend/js/features/source-editor/extensions/visual/atomic-decorations.ts
@@ -843,7 +843,7 @@ export const atomicDecorations = (options: Options) => {
           if (shouldDecorate(state, nodeRef)) {
             const line = state.doc.lineAt(nodeRef.from)
             const from = extendBackwardsOverEmptyLines(state.doc, line)
-            const to = extendForwardsOverEmptyLines(state.doc, line)
+            const { to } = state.doc.lineAt(nodeRef.to)
 
             if (shouldDecorate(state, { from, to })) {
               decorations.push(