diff --git a/assets/js/hooks/session.js b/assets/js/hooks/session.js
index 7b52a4768..a5ece8a33 100644
--- a/assets/js/hooks/session.js
+++ b/assets/js/hooks/session.js
@@ -1064,17 +1064,17 @@ const Session = {
 
   isCell(focusableId) {
     const el = this.getFocusableEl(focusableId);
-    return el.dataset.element === "cell";
+    return el.hasAttribute("data-el-cell");
   },
 
   isSection(focusableId) {
     const el = this.getFocusableEl(focusableId);
-    return el.dataset.element === "section-headline";
+    return el.hasAttribute("data-el-section-headline");
   },
 
   isNotebook(focusableId) {
     const el = this.getFocusableEl(focusableId);
-    return el.dataset.element === "notebook-headline";
+    return el.hasAttribute("data-el-notebook-headline");
   },
 
   getFocusableEl(focusableId) {