diff --git a/src/Umbraco.Web.UI.Client/src/packages/block/block-rte/tiptap-extension/block-picker.extension.ts b/src/Umbraco.Web.UI.Client/src/packages/block/block-rte/tiptap-extension/block-picker.extension.ts
index d70c9e9adc..42e48b8a7d 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/block/block-rte/tiptap-extension/block-picker.extension.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/block/block-rte/tiptap-extension/block-picker.extension.ts
@@ -122,10 +122,7 @@ export default class UmbTiptapBlockPickerExtension extends UmbTiptapToolbarEleme
 	}
 
 	override isActive(editor: Editor) {
-		return (
-			editor.isActive(`umb-rte-block[${UMB_DATA_CONTENT_UDI}]`) ||
-			editor.isActive(`umb-rte-block-inline[${UMB_DATA_CONTENT_UDI}]`)
-		);
+		return editor.isActive('umbRteBlock') || editor.isActive('umbRteBlockInline');
 	}
 
 	override async execute() {