1 files changed, 5 insertions, 5 deletions
diff --git a/src/client/views/nodes/EquationBox.tsx b/src/client/views/nodes/EquationBox.tsx
index 576b5bbe0..dcc6e27ed 100644
--- a/src/client/views/nodes/EquationBox.tsx
+++ b/src/client/views/nodes/EquationBox.tsx
@@ -30,12 +30,12 @@ export class EquationBox extends ViewBoxBaseComponent<FieldViewProps>() {
 
     componentDidMount() {
         this._props.setContentViewBox?.(this);
-        if (Doc.SelectOnLoad === this.Document && (!DocumentView.LightboxDoc() || DocumentView.LightboxContains(this.DocumentView?.()))) {
+        if (DocumentView.SelectOnLoad === this.Document && (!DocumentView.LightboxDoc() || DocumentView.LightboxContains(this.DocumentView?.()))) {
             this._props.select(false);
 
-            this._ref.current!.mathField.focus();
-            this.dataDoc.text === 'x' && this._ref.current!.mathField.select();
-            Doc.SetSelectOnLoad(undefined);
+            this._ref.current?.mathField.focus();
+            this.dataDoc.text === 'x' && this._ref.current?.mathField.select();
+            DocumentView.SetSelectOnLoad(undefined);
         }
         reaction(
             () => this._props.isSelected(),
@@ -66,7 +66,7 @@ export class EquationBox extends ViewBoxBaseComponent<FieldViewProps>() {
                 color: StrCast(this.Document.color),
                 fontSize: this.fontSize,
             });
-            Doc.SetSelectOnLoad(nextEq);
+            DocumentView.SetSelectOnLoad(nextEq);
             this._props.addDocument?.(nextEq);
             e.stopPropagation();
         }